【debug】RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

 Error cudnn problem:

return F.conv2d(input, weight, bias, self.stride,
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
You can try to repro this exception using the following code snippet. If that doesn't trigger the error, please include your original repro script when reporting this issue.

I always report an error here in net.forward(). It may be an environmental problem, but this environment is used well. The cudnn version corresponds to torch, but an error is reported. I tried several methods, and finally solved it with a simple command.

solve:

CUDA_VISIBLE_DEVICES=1 python inference.py

Just add a line to specify the GPU number, perfect! ! !

Useful Don't forget to click three times! ! !