I am getting this error with:
Ubuntu 16.04
torch     1.8.0+cu111
Python 3.7.10
NVIDIA-SMI 450.80.02
Driver Version: 450.80.02
CUDA Version: 11.0
CUDNN_VERSION 7501
Here’s 2 the entire Conda environment.
Additional information:
The error occurs for me during the loss.backward() call.
Not sure why this error occurs somewhere in the middle of the epoch. That means it works fine sometimes and sometimes not. Any suggestions/ comments are welcome.