Skip to content

Commit 61fd271

Browse files
committed
Add cuda 12 wheels for 3.11, 3.12, and Linux
1 parent b996c18 commit 61fd271

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

pyproject.toml

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,18 @@ cuda = [
3333
"onnxruntime-gpu"
3434
]
3535
cuda-12 = [
36-
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/2692857e-05ef-43b4-ba9c-ccf1c22c437c/_packaging/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/download/onnxruntime-gpu/1.17.1/onnxruntime_gpu-1.17.1-cp310-cp310-win_amd64.whl ; platform_system == 'Windows'"
36+
# CUDA 12, Python 3.10, Windows
37+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp310-cp310-win_amd64.whl/content; platform_system == 'Windows'; python_version == '3.10'",
38+
# CUDA 12, Python 3.10, Linux
39+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp311-cp311-manylinux_2_28_x86_64.whl/content; platform_system == 'Linux'; python_version == '3.10'",
40+
# CUDA 12, Python 3.11, Windows
41+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp311-cp311-win_amd64.whl/content; platform_system == 'Windows'; python_version == '3.11'",
42+
# CUDA 12, Python 3.11, Linux
43+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp311-cp311-manylinux_2_28_x86_64.whl/content; platform_system == 'Linux'; python_version == '3.11'",
44+
# CUDA 12, Python 3.12, Windows
45+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp312-cp312-win_amd64.whl/content; platform_system == 'Windows'; python_version == '3.12'",
46+
# CUDA 12, Python 3.12, Linux
47+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp312-cp312-manylinux_2_28_x86_64.whl/content; platform_system == 'Linux'; python_version == '3.12'"
3748
]
3849
directml = [
3950
"onnxruntime-directml"

0 commit comments

Comments
 (0)