@@ -79,8 +79,7 @@ http_archive(
7979 sha256 = "7f3fbe6201708de409532a32d647af6b4bdb10d7f045d557270549e286487289" ,
8080 strip_prefix = "cudnn-linux-x86_64-8.4.1.114_cuda11.4-archive" ,
8181 urls = [
82- "http://cuda-repo/release-candidates/kitbundles/cudnn/v8.4_cuda_11.4/8.4.1.114/redist/cudnn/cudnn/linux-x86_64/cudnn-linux-x86_64-8.4.1.114_cuda11.4-archive.tar.xz" ,
83- #"https://developer.nvidia.com/compute/cudnn/secure/8.4.1/local_installers/11.6/cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz",
82+ "https://developer.nvidia.com/compute/cudnn/secure/8.4.1/local_installers/11.6/cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz" ,
8483 ],
8584)
8685
@@ -90,8 +89,7 @@ http_archive(
9089 sha256 = "8107861af218694130f170e071f49814fa3e27f1386ce7cb6d807ac05a7fcf0e" ,
9190 strip_prefix = "TensorRT-8.4.1.5" ,
9291 urls = [
93- "http://cuda-repo/release-candidates/Libraries/TensorRT/v8.4/8.4.1.5-01a2da81/11.6-r510/Linux-x64-agnostic/tar/TensorRT-8.4.1.5.Linux.x86_64-gnu.cuda-11.6.cudnn8.4.tar.gz" ,
94- #"https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/8.4.1/tars/tensorrt-8.4.1.5.linux.x86_64-gnu.cuda-11.6.cudnn8.4.tar.gz",
92+ "https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/8.4.1/tars/tensorrt-8.4.1.5.linux.x86_64-gnu.cuda-11.6.cudnn8.4.tar.gz" ,
9593 ],
9694)
9795
@@ -130,20 +128,11 @@ http_archive(
130128# build_file = "@//third_party/tensorrt/local:BUILD"
131129#)
132130
133- # #########################################################################
134- # # Testing Dependencies (optional - comment out on aarch64)
135- # #########################################################################
136- # pip_install(
137- # name = "torch_tensorrt_py_deps",
138- # requirements = "//py:requirements.txt",
139- # )
140-
141- # pip_install(
142- # name = "py_test_deps",
143- # requirements = "//tests/py:requirements.txt",
144- # )
131+ #########################################################################
132+ # Development Dependencies (optional - comment out on aarch64)
133+ #########################################################################
145134
146135pip_install (
147- name = "pylinter_deps " ,
148- requirements = "//tools/linter :requirements.txt" ,
136+ name = "devtools_deps " ,
137+ requirements = "//:requirements-dev .txt" ,
149138)
0 commit comments