My environment is python3.6.9 windows10 torch1.2.0. cuda9.2
My problem is when I running FC network the code works well in both CPU and GPU.
But when it comes to CNN, I can only train it on CPU. It raises an error when I try to train it on GPU.
like that: Process finished with exit code -1073741819 (0xC0000005)
I find the error raised when the code goes to loss.backword.
Please help me. Thank you a lot!
The error happened when I use the first column instead of the second.
device = torch.device(“cuda:0”)
device = torch.device(“cuda:0” if opt.cuda else “cpu”)