转自: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
本文解决了一个常见的Pytorch CUDA运行时错误:CUDAerror:unknownerror。通过在导入torch后调用torch.cuda.current_device(),可以避免这个错误,确保GPU资源正确初始化。

2万+

被折叠的 条评论
为什么被折叠?



