saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix Python Packaging Pipeline (Training Torch 1.9.0 Cuda 11.4) (#8738)

Browse source
This commit is contained in:
Changming Sun 2021-08-16 14:46:43 -07:00 committed by GitHub
parent 9cefd1303b
commit 8335d3dc0b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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/azure-pipelines/orttraining-py-packaging-pipeline-torch190-cuda114.yml
View file

@ -6,7 +6,7 @@ stages:
build_py_parameters: --enable_training --update --build
torch_version: '1.9.0'
cuda_version: '11.4'
gcc_version: 9
gcc_version: 10
cmake_cuda_architectures: 37;50;52;60;61;70;75;80;86
docker_file: Dockerfile.manylinux2014_training_cuda11_4
agent_pool: Onnxruntime-Linux-GPU