When I use Conv1d or Conv2d layers on pytorch, the process is killed unexpectedly. I am getting the error in the following line:
loss.backward()
My set up:
- Windows 10
- cuda 10.2
- cudnn 7.6.5
- RTX 2060 Super
- Nvidia driver 451.67
- Pycharm 2020.04
Error:
Process finished with exit code -1073741819 (0xC0000005)
In comparison, when I replace the conv layer with a dense layer, the problem doesn't occur.
For more comparison, the same project and the same code was run on Ubuntu 20.04 as well, and it worked quite well.