RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

[chehefei]

I have the following problems: RuntimeError: Unable to find a valid cuDNN algorithm to run convolution I changed the batch_size from 128 to 32 or less, but it still doesn't work. How should I solve it? I would appreciate your reply Here are the details: ——————————————————- epoch-> 2999 Iteration-> 0 , Iteration_time-> 1.6496 , Iteration_loss 0.0034 Epoch 2999 , average loss 0.0034224344417452812 , epoch_time 1.6496999263763428 Saving Model Done saving Model Defining model Only Conv model trained Number of epochs 200

epoch-> 0 Traceback (most recent call last): File "main.py", line 368, in train_conv(args) File "main.py", line 334, in train_conv loss.backward() File "/home/menhaojie/anaconda3/envs/pytorch35/lib/python3.5/site-packages/torch/tensor.py", line 198, in backward torch.autograd.backward(self, gradient, retain_graph, create_graph) File "/home/menhaojie/anaconda3/envs/pytorch35/lib/python3.5/site-packages/torch/autograd/init.py", line 100, in backward allow_unreachable=True) # allow_unreachable flag RuntimeError: Unable to find a valid cuDNN algorithm to run convolution (try_all at /opt/conda/conda-bld/pytorch_1591914868920/work/aten/src/ATen/native/cudnn/Conv.cpp:693)