2022-12-30 11:53:17 -05:00
|
|
|
# NVidia TensorRT Support (amd64 only)
|
2023-07-06 15:20:33 -04:00
|
|
|
--extra-index-url 'https://pypi.nvidia.com'
|
2025-07-07 17:12:52 -06:00
|
|
|
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'
|
2024-10-24 07:48:14 -06:00
|
|
|
onnx==1.16.*; platform_machine == 'x86_64'
|
2025-05-30 15:30:21 -06:00
|
|
|
onnxruntime-gpu==1.22.*; platform_machine == 'x86_64'
|
2024-09-15 19:30:30 -06:00
|
|
|
protobuf==3.20.3; platform_machine == 'x86_64'
|
