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