训练出错

[yuecaihan]

Traceback (most recent call last):
File "/root/autodl-tmp/H-vmunet/train.py", line 188, in
main(config)
File "/root/autodl-tmp/H-vmunet/train.py", line 131, in main
train_one_epoch(
File "/root/autodl-tmp/H-vmunet/engine.py", line 40, in train_one_epoch
loss.backward()
File "/root/miniconda3/envs/vmunet/lib/python3.8/site-packages/torch/_tensor.py", line 487, in backward
torch.autograd.backward(
File "/root/miniconda3/envs/vmunet/lib/python3.8/site-packages/torch/autograd/init.py", line 197, in backward
Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
RuntimeError: cuDNN error: CUDNN_STATUS_MAPPING_ERROR

[SKYMICHEAL]

问题解决了吗，我也是这个错误