remove cudnn dependency

Author: zewenli98

README.md

@@ -118,7 +118,6 @@ These are the following dependencies used to verify the testcases. Torch-TensorR
 - Bazel 5.2.0
 - Libtorch 2.4.0.dev (latest nightly) (built with CUDA 12.1)
 - CUDA 12.1
-- cuDNN 8.9.5
 - TensorRT 10.0.1.6
 
 ## Prebuilt Binaries and Wheel files

WORKSPACE

@@ -101,12 +101,6 @@ http_archive(
 #    build_file = "third_party/libtorch/BUILD"
 #)
 
-#new_local_repository(
-#    name = "cudnn",
-#    path = "/usr/",
-#    build_file = "@//third_party/cudnn/local:BUILD"
-#)
-
 #new_local_repository(
 #   name = "tensorrt",
 #   path = "/usr/",

dev_dep_versions.yml

@@ -1,4 +1,3 @@
 __version__: "2.3.0.dev0"
 __cuda_version__: "12.1"
-__cudnn_version__: "8.9"
-__tensorrt_version__: "10.0.0.6"
+__tensorrt_version__: "10.0.1.6"

py/requirements.txt

@@ -1,9 +1,9 @@
 numpy
 packaging
 pybind11==2.6.2
---extra-index-url https://download.pytorch.org/whl/nightly/cu121
-torch>=2.4.0.dev,<2.5.0
-torchvision>=0.19.0.dev,<0.20.0
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.3.0
+torchvision==0.18.0
 --extra-index-url https://pypi.ngc.nvidia.com
 pyyaml
-tensorrt
+tensorrt==10.0.1.6