Skip to content

Commit 14543de

Browse files
baijumeswanirachguo
authored and
rachguo
committed
Remove cuda gencode 90 to reduce onnxruntime-training package size (#19486)
1 parent e02b783 commit 14543de

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cuda.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ stages:
1515
torch_version: '2.0.0'
1616
opset_version: '15'
1717
cuda_version: '11.8'
18-
cmake_cuda_architectures: 60;61;70;75;80;86;90
18+
cmake_cuda_architectures: 60;61;70;75;80;86
1919
docker_file: Dockerfile.manylinux2_28_training_cuda11_8
2020
agent_pool: Onnxruntime-Linux-GPU
2121
upload_wheel: 'yes'

0 commit comments

Comments
 (0)