diff --git a/extension_cpp/__init__.py b/extension_cpp/__init__.py index 769c697..bbcf859 100644 --- a/extension_cpp/__init__.py +++ b/extension_cpp/__init__.py @@ -1,2 +1,6 @@ import torch +inside_virtual_environment = sys.prefix != sys.base_prefix +if inside_virtual_environment and os.name == 'nt': + dll_dir = os.path.join(sys.prefix, 'Lib\\site-packages\\torch\\lib') + handle = os.add_dll_directory(dll_dir) from . import _C, ops
Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.
Alternative Proxies: