feat: remove unnecessary direct loading cudnn

narendasan · narendasan · commit 9149659405dc · 2022-07-22T16:49:26.000-07:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
diff --git a/py/torch_tensorrt/__init__.py b/py/torch_tensorrt/__init__.py
@@ -44,9 +44,6 @@ def _find_lib(name, paths):
 
     if sys.platform.startswith("win"):
         WIN_LIBS = [
-            f"cublas64_{CUDA_MAJOR}.dll",
-            f"cublasLt64_{CUDA_MAJOR}.dll",
-            f"cudnn64_{CUDNN_MAJOR}.dll",
             "nvinfer.dll",
             "nvinfer_plugin.dll",
         ]
@@ -76,7 +73,6 @@ def _find_lib(name, paths):
             ]
 
         LINUX_LIBS = [
-            f"libcudnn.so.{CUDNN_MAJOR}",
             f"libnvinfer.so.{TENSORRT_MAJOR}",
             f"libnvinfer_plugin.so.{TENSORRT_MAJOR}",
         ]

Original file line number	Diff line number	Diff line change
`@@ -44,9 +44,6 @@ def _find_lib(name, paths):`
`44`	`44`
`45`	`45`	`if sys.platform.startswith("win"):`
`46`	`46`	`WIN_LIBS = [`
`47`		`- f"cublas64_{CUDA_MAJOR}.dll",`
`48`		`- f"cublasLt64_{CUDA_MAJOR}.dll",`
`49`		`- f"cudnn64_{CUDNN_MAJOR}.dll",`
`50`	`47`	`"nvinfer.dll",`
`51`	`48`	`"nvinfer_plugin.dll",`
`52`	`49`	`]`
`@@ -76,7 +73,6 @@ def _find_lib(name, paths):`
`76`	`73`	`]`
`77`	`74`
`78`	`75`	`LINUX_LIBS = [`
`79`		`- f"libcudnn.so.{CUDNN_MAJOR}",`
`80`	`76`	`f"libnvinfer.so.{TENSORRT_MAJOR}",`
`81`	`77`	`f"libnvinfer_plugin.so.{TENSORRT_MAJOR}",`
`82`	`78`	`]`