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

retarget torch181 to torch182 (#8947)

Browse source 
Co-authored-by: liqun <liqun@OrtTrainingDev4.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
This commit is contained in:
liqun Fu 2021-09-03 09:44:42 -07:00 committed by GitHub
parent 0cc2909573
commit a7f5bd226b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 14 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-torch181-cuda102.yml → tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-torch182-cuda102.yml
View file

@ -4,7 +4,7 @@ stages:
- template: templates/py-packaging-training-cuda-stage.yml
parameters:
build_py_parameters: --enable_training --update --build
torch_version: '1.8.1'
torch_version: '1.8.2'
cuda_version: '10.2'
gcc_version: 8
cmake_cuda_architectures: 35;37;50;52;60;61;70

2
tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-torch181-cuda111.yml → tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-torch182-cuda111.yml
View file

@ -4,7 +4,7 @@ stages:
- template: templates/py-packaging-training-cuda-stage.yml
parameters:
build_py_parameters: --enable_training --update --build
torch_version: '1.8.1'
torch_version: '1.8.2'
cuda_version: '11.1'
gcc_version: 9
cmake_cuda_architectures: 37;50;52;60;61;70;75;80

6
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch1.8.1_cu10.2.txt
View file

@ -1,6 +0,0 @@
--pre
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.8.1+cu102
torchvision==0.9.1+cu102
torchtext==0.9.1
setuptools>=41.4.0

6
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch1.8.1_cu11.1.txt
View file

@ -1,6 +0,0 @@
--pre
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.8.1+cu111
torchvision==0.9.1+cu111
torchtext==0.9.1
setuptools>=41.4.0

6
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch1.8.2_cu10.2.txt Normal file
View file

@ -0,0 +1,6 @@
--pre
-f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
torch==1.8.2+cu102
torchvision==0.9.2+cu102
torchtext==0.9.2
setuptools>=41.4.0

6
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch1.8.2_cu11.1.txt Normal file
View file

@ -0,0 +1,6 @@
--pre
-f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
torch==1.8.2+cu111
torchvision==0.9.2+cu111
torchtext==0.9.2
setuptools>=41.4.0