如何解决PyTorch收到“ CUDA错误:设备端断言已触发”-如何解决?
当尝试在GPU上运行PyTorch时,我收到一条奇怪的神秘错误消息:“设备端断言已触发”。
例如,如果我键入
import torch
torch.randn(1,device='cuda')
我得到:
RuntimeError:CUDA错误:设备端断言已触发
如何解决此错误?键入torch.cuda.is_available()
将返回True
,并且重新启动计算机也无济于事。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。