chore: Update cudnn depedency

peri044 · peri044 · commit df323df9eda9 · 2022-06-30T15:56:55.000-07:00
Signed-off-by: Dheeraj Peri &lt;peri.dheeraj@gmail.com&gt;
diff --git a/WORKSPACE b/WORKSPACE
@@ -76,10 +76,10 @@ http_archive(
 http_archive(
     name = "cudnn",
     build_file = "@//third_party/cudnn/archive:BUILD",
-    sha256 = "0e5d2df890b9967efa6619da421310d97323565a79f05a1a8cb9b7165baad0d7",
+    sha256 = "ec96d2376d81fca42bdd3d4c3d705a99b29a065bab57f920561c763e29c67d01",
     strip_prefix = "cuda",
     urls = [
-        "https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.2.4/11.4_20210831/cudnn-11.4-linux-x64-v8.2.4.15.tgz",
+        "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",
     ],
 )
 
diff --git a/third_party/cudnn/archive/BUILD b/third_party/cudnn/archive/BUILD
@@ -9,7 +9,7 @@ cc_library(
 
 cc_import(
     name = "cudnn_lib",
-    shared_library = "lib64/libcudnn.so",
+    shared_library = "lib/libcudnn.so",
     visibility = ["//visibility:private"],
 )
 

Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ cc_library(`
`9`	`9`
`10`	`10`	`cc_import(`
`11`	`11`	`name = "cudnn_lib",`
`12`		`- shared_library = "lib64/libcudnn.so",`
	`12`	`+ shared_library = "lib/libcudnn.so",`
`13`	`13`	`visibility = ["//visibility:private"],`
`14`	`14`	`)`
`15`	`15`