Update library loading for tensorrt (#5087)

* Update library loading for tensorrt

* Add symlink to libnvrtc
This commit is contained in:
Nate Meyer
2023-01-14 14:14:27 -05:00
committed by GitHub
parent 19d17c8c81
commit 60b2315028
4 changed files with 10 additions and 8 deletions

View File

@@ -5,4 +5,5 @@ cuda-python == 11.7; platform_machine == 'x86_64'
cython == 0.29.*; platform_machine == 'x86_64'
nvidia-cuda-runtime-cu11 == 11.7.*; platform_machine == 'x86_64'
nvidia-cublas-cu11 == 11.11.*; platform_machine == 'x86_64'
nvidia-cudnn-cu11 == 8.7.*; platform_machine == 'x86_64'
nvidia-cudnn-cu11 == 8.7.*; platform_machine == 'x86_64'
nvidia-cuda-nvrtc-cu11 == 11.7.*; platform_machine == 'x86_64'