RuntimeError: cuDNN error: CUDNN_STATUS_MAPPING_ERROR

[huangjinghuangjing]

trafficstars

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