blob: 0f1b400e234dede667ed3bb832b200acb4b3d621 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
--- a/torch/__init__.py 2023-04-05 11:12:25.682681130 +0200
+++ b/torch/__init__.py 2023-04-05 11:13:00.640170307 +0200
@@ -165,7 +165,7 @@
lib_path = os.path.join(os.path.dirname(here), 'lib', lib_name)
try:
- ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
+ ctypes.CDLL(lib_name, mode=ctypes.RTLD_GLOBAL)
except OSError as err:
# Can only happen for wheel with cuda libs as PYPI deps
# As PyTorch is not purelib, but nvidia-*-cu11 is
@@ -187,7 +187,7 @@
raise err
for lib_folder, lib_name in cuda_libs.items():
_preload_cuda_deps(lib_folder, lib_name)
- ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
+ ctypes.CDLL(lib_name, mode=ctypes.RTLD_GLOBAL)
if (USE_RTLD_GLOBAL_WITH_LIBTORCH or os.getenv('TORCH_USE_RTLD_GLOBAL')) and \
|
