Compare commits

...
Author SHA1 Message Date
Nicolas Mowen 4233000d13 Update 2026-03-08 08:06:43 -06:00
Nicolas Mowen c553cc1964 Update support for newer Nvidia GPUs 2026-03-08 07:54:16 -06:00
2 changed files with 30 additions and 14 deletions
+16
View File
@@ -0,0 +1,16 @@
{
"permissions": {
"allow": [
"WebFetch(domain:onnxruntime.ai)",
"WebFetch(domain:github.com)",
"WebFetch(domain:pypi.org)",
"Bash(pip index versions onnxruntime-gpu)",
"Bash(pip install onnxruntime-gpu==1.24.3 --dry-run)",
"WebFetch(domain:raw.githubusercontent.com)",
"Bash(pip download onnxruntime-gpu==1.24.3 --no-deps -d /tmp/ort-check --python-version 3.12 --platform manylinux_2_28_x86_64)",
"WebFetch(domain:files.pythonhosted.org)",
"Bash(gh api repos/microsoft/onnxruntime/contents/docs/CUDA_GPU_Compatibility.md --jq '.content')",
"Bash(gh search code \"cuda 13\" --repo microsoft/onnxruntime --limit 5)"
]
}
}
+14 -14
View File
@@ -1,18 +1,18 @@
# NVidia TensorRT Support (amd64 only)
# Nvidia ONNX Runtime GPU Support
--extra-index-url 'https://pypi.nvidia.com'
cython==3.0.*; platform_machine == 'x86_64'
nvidia_cuda_cupti_cu12==12.5.82; platform_machine == 'x86_64'
nvidia-cublas-cu12==12.5.3.*; platform_machine == 'x86_64'
nvidia-cudnn-cu12==9.3.0.*; platform_machine == 'x86_64'
nvidia-cufft-cu12==11.2.3.*; platform_machine == 'x86_64'
nvidia-curand-cu12==10.3.6.*; platform_machine == 'x86_64'
nvidia_cuda_nvcc_cu12==12.5.82; platform_machine == 'x86_64'
nvidia-cuda-nvrtc-cu12==12.5.82; platform_machine == 'x86_64'
nvidia_cuda_runtime_cu12==12.5.82; platform_machine == 'x86_64'
nvidia_cusolver_cu12==11.6.3.*; platform_machine == 'x86_64'
nvidia_cusparse_cu12==12.5.1.*; platform_machine == 'x86_64'
nvidia_nccl_cu12==2.23.4; platform_machine == 'x86_64'
nvidia_nvjitlink_cu12==12.5.82; platform_machine == 'x86_64'
nvidia-cuda-cupti-cu12==12.9.79; platform_machine == 'x86_64'
nvidia-cublas-cu12==12.9.1.*; platform_machine == 'x86_64'
nvidia-cudnn-cu12==9.19.0.*; platform_machine == 'x86_64'
nvidia-cufft-cu12==11.4.1.*; platform_machine == 'x86_64'
nvidia-curand-cu12==10.3.10.*; platform_machine == 'x86_64'
nvidia-cuda-nvcc-cu12==12.9.86; platform_machine == 'x86_64'
nvidia-cuda-nvrtc-cu12==12.9.86; platform_machine == 'x86_64'
nvidia-cuda-runtime-cu12==12.9.79; platform_machine == 'x86_64'
nvidia-cusolver-cu12==11.7.5.*; platform_machine == 'x86_64'
nvidia-cusparse-cu12==12.5.10.*; platform_machine == 'x86_64'
nvidia-nccl-cu12==2.29.7; platform_machine == 'x86_64'
nvidia-nvjitlink-cu12==12.9.86; platform_machine == 'x86_64'
onnx==1.16.*; platform_machine == 'x86_64'
onnxruntime-gpu==1.22.*; platform_machine == 'x86_64'
onnxruntime-gpu==1.24.*; platform_machine == 'x86_64'
protobuf==3.20.3; platform_machine == 'x86_64'