RuntimeError: cuDNN error: CUDNN_STATUS_MAPPING_ERROR

[huangjinghuangjing]

When I run the train.py occur this problem, anyone can solve this?

Traceback (most recent call last):
File "/home/common1/huangjing/MyCode/PythonCode/pytorch-video-recognition/train.py", line 201, in
train_model()
File "/home/common1/huangjing/MyCode/PythonCode/pytorch-video-recognition/train.py", line 138, in train_model
loss.backward()
File "/home/huangjing/miniconda3/envs/PyCharm/lib/python3.7/site-packages/torch/tensor.py", line 150, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File "/home/huangjing/miniconda3/envs/PyCharm/lib/python3.7/site-packages/torch/autograd/init.py", line 99, in backward
allow_unreachable=True) # allow_unreachable flag
RuntimeError: cuDNN error: CUDNN_STATUS_MAPPING_ERROR