Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Load all DLLs in the lib directory for Windows (v.1.5.0)
  • Loading branch information
peterjc123 committed Mar 25, 2020
1 parent fb59a9c commit d4d7160
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions torch/__init__.py
Expand Up @@ -71,6 +71,11 @@

os.environ['PATH'] = ';'.join(dll_paths)

import glob
dlls = glob.glob(os.path.join(th_dll_path, '*.dll'))
for dll in dlls:
ctypes.CDLL(dll)


# See Note [Global dependencies]
def _load_global_deps():
Expand Down

0 comments on commit d4d7160

Please sign in to comment.