关于SPIN网络训练过程中产生的问题

[xiaojiangjiangjinger]

Traceback (most recent call last):
File "/root/.pycharm_helpers/pydev/pydevd.py", line 1477, in _exec
pydev_imports.execfile(file, globals, locals) # execute the script
File "/root/.pycharm_helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile
exec(compile(contents+"\n", file, 'exec'), glob, loc)
File "/mnt/SPIN/train.py", line 123, in
sr = model(lr)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 1015, in _call_impl
return forward_call(*input, **kwargs)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/parallel/data_parallel.py", line 165, in forward
return self.module(*inputs[0], **kwargs[0])
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 1015, in _call_impl
return forward_call(*input, **kwargs)
File "/mnt/SPIN/models/spin.py", line 54, in forward
x /= 255.
RuntimeError: CUDA error: no kernel image is available for execution on the device
运行环境：
NVIDIA 4090
CUDA 10.1
pytorch 1.9.0
python 3.6.9

已经尝试过替换CUDA版本、pytorch版本，依然不能解决该问题，麻烦作者给予一些解决思路，我将会不胜感激。