Моя проблема в том, что когда я запускаю сеть FC, код хорошо работает как в CPU, так и в GPU. Но когда дело доходит до CNN, я могу тренировать его только на процессоре. Вызывает ошибку, когда я пытаюсь обучить его на GPU.
Примерно так:
Процесс завершен с кодом выхода -1073741819 (0xC0000005)
Я нахожу ошибку, возникающую, когда код переходит к loss.backword. Ошибка произошла, когда я использую первый столбец вместо второго.
device = torch.device("cuda:0")
device = torch.device("cuda:0" if opt.cuda else "cpu")
Моя среда - Python 3.6.9, Windows 10, Torch 1.2.0, Cuda 9.2.