训练的时候报错

[h1x1h]

No CUDA runtime is found, using CUDA_HOME='D:\CUDA\cuda'
Using C:\Users\Administrator\AppData\Local\torch_extensions\torch_extensions\Cache\py39_cu118 as PyTorch extensions root...
Detected CUDA files, patching ldflags
Emitting ninja build file C:\Users\Administrator\AppData\Local\torch_extensions\torch_extensions\Cache\py39_cu118\wkv\build.ninja...
Traceback (most recent call last):
File "D:\LR_HR\Restore-RWKV-main\main_train_Restore_RWKV.py", line 3, in
from model.Restore_RWKV import Restore_RWKV
File "D:\LR_HR\Restore-RWKV-main\model\Restore_RWKV.py", line 15, in
wkv_cuda = load(
File "D:\Anaconda3\envs\rwkv_env\lib\site-packages\torch\utils\cpp_extension.py", line 1284, in load
return _jit_compile(
File "D:\Anaconda3\envs\rwkv_env\lib\site-packages\torch\utils\cpp_extension.py", line 1509, in _jit_compile
_write_ninja_file_and_build_library(
File "D:\Anaconda3\envs\rwkv_env\lib\site-packages\torch\utils\cpp_extension.py", line 1611, in _write_ninja_file_and_build_library
_write_ninja_file_to_build_library(
File "D:\Anaconda3\envs\rwkv_env\lib\site-packages\torch\utils\cpp_extension.py", line 2007, in _write_ninja_file_to_build_library
cuda_flags = common_cflags + COMMON_NVCC_FLAGS + _get_cuda_arch_flags()
File "D:\Anaconda3\envs\rwkv_env\lib\site-packages\torch\utils\cpp_extension.py", line 1773, in _get_cuda_arch_flags
arch_list[-1] += '+PTX'
IndexError: list index out of range请问这个报错改怎么解决 呢？训练了MRI超分任务