Skip to content

使用--cuda_on 1启用CUDA编译时提示找不到模块 #4

Description

@leimm2002

这是当时的报错信息:

Using C:\Users\Momozawa Chiyuki\AppData\Local\torch_extensions\torch_extensions\Cache\py310_cu117 as PyTorch extensions root...
Detected CUDA files, patching ldflags
Emitting ninja build file C:\Users\Momozawa Chiyuki\AppData\Local\torch_extensions\torch_extensions\Cache\py310_cu117\wkv_cuda\build.ninja...
Building extension module wkv_cuda...
Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
ninja: no work to do.
Loading extension module wkv_cuda...
Traceback (most recent call last):
File "C:\RWKV\RWKV_Role_Playing\webui.py", line 17, in <module>
from modules.model_utils import ModelUtils
File "C:\RWKV\RWKV_Role_Playing\modules\model_utils.py", line 6, in <module>
from rwkv.model import RWKV
File "C:\Users\Momozawa Chiyuki\AppData\Local\Programs\Python\Python310\lib\site-packages\rwkv\model.py", line 29, in <module>
load(
File "C:\Users\Momozawa Chiyuki\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\utils\cpp_extension.py", line 1284, in load
return _jit_compile(
File "C:\Users\Momozawa Chiyuki\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\utils\cpp_extension.py", line 1535, in _jit_compile
return _import_module_from_library(name, build_directory, is_python_module)
File "C:\Users\Momozawa Chiyuki\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\utils\cpp_extension.py", line 1934, in _import_module_from_library
torch.ops.load_library(filepath)
File "C:\Users\Momozawa Chiyuki\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\_ops.py", line 643, in load_library
ctypes.CDLL(path)
File "C:\Users\Momozawa Chiyuki\AppData\Local\Programs\Python\Python310\lib\ctypes\__init__.py", line 374, in __init__    self._handle = _dlopen(self._name, mode)
FileNotFoundError: Could not find module 'C:\Users\Momozawa Chiyuki\AppData\Local\torch_extensions\torch_extensions\Cache\py310_cu117\wkv_cuda\wkv_cuda.pyd' (or one of its dependencies). Try using the full path with constructor syntax.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions