2022-12-28 17:39:14 +03:00
|
|
|
# NVidia TensorRT Support (amd64 only)
|
2022-12-12 09:40:42 +03:00
|
|
|
nvidia-pyindex; platform_machine == 'x86_64'
|
|
|
|
|
nvidia-tensorrt == 8.4.1.5; platform_machine == 'x86_64'
|
|
|
|
|
cuda-python == 11.7; platform_machine == 'x86_64'
|
|
|
|
|
cython == 0.29.*; platform_machine == 'x86_64'
|
2022-12-28 17:39:14 +03:00
|
|
|
nvidia-cuda-runtime-cu11 == 2022.4.25; platform_machine == 'x86_64'
|
|
|
|
|
nvidia-cuda-runtime-cu117 == 11.7.*; platform_machine == 'x86_64'
|
|
|
|
|
nvidia-cublas-cu11 == 2022.4.8; platform_machine == 'x86_64'
|
|
|
|
|
nvidia-cublas-cu117 == 11.10.*; platform_machine == 'x86_64'
|
|
|
|
|
nvidia-cudnn-cu11 == 2022.5.19; platform_machine == 'x86_64'
|
|
|
|
|
nvidia-cudnn-cu116 == 8.4.*; platform_machine == 'x86_64'
|