pytorch 报错:
RuntimeError: copy_if failed to synchronize: cudaErrorIllegalAddress: an illegal memory access was encountered
我的出错原因是输入数据忘记加.to(device)或.cuda()
pytorch 报错:
RuntimeError: copy_if failed to synchronize: cudaErrorIllegalAddress: an illegal memory access was encountered
我的出错原因是输入数据忘记加.to(device)或.cuda()