Update requirements-CUDA-Linux.txt

TNTwise · May 8, 2024 · 825277c · 825277c
1 parent ef3b380
commit 825277c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/requirements-CUDA-Linux.txt b/requirements-CUDA-Linux.txt
@@ -5,4 +5,9 @@ spandrel
 torchvision==0.15.2
 cupy-cuda11x
 fastrlock
+https://github.com/pytorch/TensorRT/releases/download/v1.4.0/torch_tensorrt-1.4.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
+tensorrt==8.6.1
+nvidia-cuda-runtime-cu12
+nvidia-cublas-cu12
+nvidia-cudnn-cu12