saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-01 03:45:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

pin pytorch lightning version for training CI (#21731)

Browse source 
### Description
<!-- Describe your changes. -->

Pins pytorch-lightning package to version 2.3.3 since version >=2.4.0
requires torch > 2.1.0 which is not compatible with cu118.


### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->

ORT 1.19 Release Preparation
This commit is contained in:
Prathik Rao 2024-08-13 20:04:56 -07:00 committed by GitHub
parent b92908e197
commit e32e3575d8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt
View file

@ -8,7 +8,7 @@ rsa==4.9
tensorboard==2.13.0
h5py
wget
pytorch-lightning
pytorch-lightning==2.3.3
deepspeed==0.9.0
fairscale==0.4.6
parameterized>=0.8.1