首页 > 编程知识 正文

cuda报错 RuntimeError CUDA error unknown error,cuda报错code35

时间:2023-05-04 14:02:51 阅读:255024 作者:213

转自: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

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。