转自:Pytorch 常见问题
苦恼了一下午,再次感谢作者!
cuda报错, RuntimeError: CUDA error: unknown error
File "D:\ProgramFiles\ProgramFiles\anaconda\lib\site-packages\torch\nn\modules\module.py", line 384, in convert
return t.to(device, dtype if t.is_floating_point() else None, non_blocking)
File "D:\ProgramFiles\ProgramFiles\anaconda\lib\site-packages\torch\cuda\__init__.py", line 163, in _lazy_init
torch._C._cuda_init()
RuntimeError: CUDA error: unknown error
在import后边加上下边一句:torch.cuda.current_device()
import torch
torch.cuda.current_device()
import torchvision