Traceback (most recent call last):
File "", line 1, in
File "D:\pycharm\PyCharm 2021.2\plugins\python\helpers\pydev_pydev_bundle\pydev_umd.py", line 198, in runfile
pydev_imports.execfile(filename, global_vars, local_vars) # execute the script
File "D:\pycharm\PyCharm 2021.2\plugins\python\helpers\pydev_pydev_imps_pydev_execfile.py", line 18, in execfile
exec(compile(contents+"\n", file, 'exec'), glob, loc)
File "D:/zzw_data/SRU_project/sru1.py", line 17, in
import sru
File "D:\pycharm\PyCharm 2021.2\plugins\python\helpers\pydev_pydev_bundle\pydev_import_hook.py", line 21, in do_import
module = self.system_import(name, *args, **kwargs)
File "D:\anaconda\envs\pytorch\lib\site-packages\sru_init.py", line 8, in
from .cuda_functional import *
File "D:\pycharm\PyCharm 2021.2\plugins\python\helpers\pydev_pydev_bundle\pydev_import_hook.py", line 21, in do_import
module = self._system_import(name, *args, **kwargs)
File "D:\anaconda\envs\pytorch\lib\site-packages\sru\cuda_functional.py", line 441, in
class SRU_Compute_GPU(Function):
File "D:\anaconda\envs\pytorch\lib\site-packages\sru\cuda_functional.py", line 444, in SRU_Compute_GPU
_SRU_PROG = Program(SRU_CODE, 'sru_prog.cu') # for pynvrtc >= 9.0
File "D:\anaconda\envs\pytorch\lib\site-packages\pynvrtc\compiler.py", line 49, in init
self._interface = NVRTCInterface(lib_name)
File "D:\anaconda\envs\pytorch\lib\site-packages\pynvrtc\interface.py", line 87, in init
self._load_nvrtc_lib(lib_path)
File "D:\anaconda\envs\pytorch\lib\site-packages\pynvrtc\interface.py", line 109, in _load_nvrtc_lib
self.lib = cdll.LoadLibrary(name)
File "D:\anaconda\envs\pytorch\lib\ctypes_init.py", line 426, in LoadLibrary
return self.dlltype(name)
File "D:\anaconda\envs\pytorch\lib\ctypes_init.py", line 348, in init
self._handle = _dlopen(self._name, mode)
OSError: [WinError 126] 找不到指定的模块。
sru模块找不到,引入本地的包中的类的话,需要和当前文件位于同一级目录的