@@ -79,8 +79,7 @@ http_archive(
79
79
sha256 = "7f3fbe6201708de409532a32d647af6b4bdb10d7f045d557270549e286487289" ,
80
80
strip_prefix = "cudnn-linux-x86_64-8.4.1.114_cuda11.4-archive" ,
81
81
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" ,
84
83
],
85
84
)
86
85
@@ -90,8 +89,7 @@ http_archive(
90
89
sha256 = "8107861af218694130f170e071f49814fa3e27f1386ce7cb6d807ac05a7fcf0e" ,
91
90
strip_prefix = "TensorRT-8.4.1.5" ,
92
91
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" ,
95
93
],
96
94
)
97
95
@@ -130,20 +128,11 @@ http_archive(
130
128
# build_file = "@//third_party/tensorrt/local:BUILD"
131
129
#)
132
130
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
+ #########################################################################
145
134
146
135
pip_install (
147
- name = "pylinter_deps " ,
148
- requirements = "//tools/linter :requirements.txt" ,
136
+ name = "devtools_deps " ,
137
+ requirements = "//:requirements-dev .txt" ,
149
138
)
0 commit comments