The above exception was the direct cause of the following exception:
Traceback (most recent call last):
File "D:\Github download\Diffusion-RWKV-main\train.py", line 24, in
from models_drwkv import DRWKV_models
File "D:\Github download\Diffusion-RWKV-main\models_drwkv.py", line 20, in
wkv_cuda = load(name="wkv", sources=["cuda/wkv_op.cpp", "cuda/wkv_cuda.cu"],
File "D:\anaconda3\envs\KWEY\lib\site-packages\torch\utils\cpp_extension.py", line 1309, in load
return _jit_compile(
File "D:\anaconda3\envs\KWEY\lib\site-packages\torch\utils\cpp_extension.py", line 1719, in _jit_compile
_write_ninja_file_and_build_library(
File "D:\anaconda3\envs\KWEY\lib\site-packages\torch\utils\cpp_extension.py", line 1832, in _write_ninja_file_and_build_library
_run_ninja_build(
File "D:\anaconda3\envs\KWEY\lib\site-packages\torch\utils\cpp_extension.py", line 2123, in _run_ninja_build
raise RuntimeError(message) from e
RuntimeError: Error building extension 'wkv'`
I am facing some technical issues, and I need help :(
`FAILED: wkv_cuda.cuda.o C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.5\bin\nvcc --generate-dependencies-with-compile --dependency-output wkv_cuda.cuda.o.d -Xcudafe --diag_suppress=dll_interface_conflict_dllexport_assumed -Xcudafe --diag_suppress=dll_interface_conflict_none_assumed -Xcudafe --diag_suppress=field_without_dll_interface -Xcudafe --diag_suppress=base_class_has_different_dll_interface -Xcompiler /EHsc -Xcompiler /wd4068 -Xcompiler /wd4067 -Xcompiler /wd4624 -Xcompiler /wd4190 -Xcompiler /wd4018 -Xcompiler /wd4275 -Xcompiler /wd4267 -Xcompiler /wd4244 -Xcompiler /wd4251 -Xcompiler /wd4819 -Xcompiler /MD -DTORCH_EXTENSION_NAME=wkv -DTORCH_API_INCLUDE_EXTENSION_H -ID:\anaconda3\envs\KWEY\lib\site-packages\torch\include -ID:\anaconda3\envs\KWEY\lib\site-packages\torch\include\torch\csrc\api\include -ID:\anaconda3\envs\KWEY\lib\site-packages\torch\include\TH -ID:\anaconda3\envs\KWEY\lib\site-packages\torch\include\THC "-IC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.5\include" -ID:\anaconda3\envs\KWEY\Include -D_GLIBCXX_USE_CXX11_ABI=0 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr -gencode=arch=compute_86,code=compute_86 -gencode=arch=compute_86,code=sm_86 -std=c++17 -res-usage "--maxrregcount 60" --use_fast_math -O3 "-Xptxas -O3" -DTmax=1024 -c "D:\Github download\Diffusion-RWKV-main\cuda\wkv_cuda.cu" -o wkv_cuda.cuda.o nvcc fatal : Unknown option '--maxrregcount 60' ninja: build stopped: subcommand failed. Traceback (most recent call last): File "D:\anaconda3\envs\KWEY\lib\site-packages\torch\utils\cpp_extension.py", line 2107, in _run_ninja_build subprocess.run( File "D:\anaconda3\envs\KWEY\lib\subprocess.py", line 526, in run raise CalledProcessError(retcode, process.args, subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.
The above exception was the direct cause of the following exception:
Traceback (most recent call last): File "D:\Github download\Diffusion-RWKV-main\train.py", line 24, in
from models_drwkv import DRWKV_models
File "D:\Github download\Diffusion-RWKV-main\models_drwkv.py", line 20, in
wkv_cuda = load(name="wkv", sources=["cuda/wkv_op.cpp", "cuda/wkv_cuda.cu"],
File "D:\anaconda3\envs\KWEY\lib\site-packages\torch\utils\cpp_extension.py", line 1309, in load
return _jit_compile(
File "D:\anaconda3\envs\KWEY\lib\site-packages\torch\utils\cpp_extension.py", line 1719, in _jit_compile
_write_ninja_file_and_build_library(
File "D:\anaconda3\envs\KWEY\lib\site-packages\torch\utils\cpp_extension.py", line 1832, in _write_ninja_file_and_build_library
_run_ninja_build(
File "D:\anaconda3\envs\KWEY\lib\site-packages\torch\utils\cpp_extension.py", line 2123, in _run_ninja_build
raise RuntimeError(message) from e
RuntimeError: Error building extension 'wkv'`