pytorch
diff --git a/‎torch/__init__.py
Copy file name to clipboardExpand all lines: torch/__init__.py
+3-2Lines changed: 3 additions & 2 deletions b/‎torch/__init__.py
Copy file name to clipboardExpand all lines: torch/__init__.py
+3-2Lines changed: 3 additions & 2 deletions
@@ -328,9 +328,10 @@ def _load_global_deps() -> None:
             # libtorch_global_deps.so always depends in cudart, check if its installed via wheel
             if "nvidia/cuda_runtime/lib/libcudart.so" not in _maps:
                 return
-            # If all abovementioned conditions are met, preload nvjitlink and nvrtc
-            _preload_cuda_deps("nvjitlink", "libnvJitLink.so.*[0-9]")
+            # If all above-mentioned conditions are met, preload nvrtc and nvjitlink
+            # Please note that order are important for CUDA-11.8 , as nvjitlink does not exist there
             _preload_cuda_deps("cuda_nvrtc", "libnvrtc.so.*[0-9]")
+            _preload_cuda_deps("nvjitlink", "libnvJitLink.so.*[0-9]")
         except Exception:
             pass