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

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

Cause analysis:

The batch_size is too large and there is not enough memory.

Solution:

Turn down batch_size.