cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

[Huiimin5]

Hi thank you for your contribution.
The problem I have when running the MNIST_example.ipynb is that it gives me a runtime error: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED.
I install the same environment except torchvision, which gives me this error:

so I changed it to: - torchvision=0.2.1.
Then the environment can be installed correctly but I get cuDNN error when running jupyter notebook.
Is it possible to know from you how to solve this problem?