【bug】RuntimeError: CUDA error: no kernel image is available for execution on the device 解决方案
使用demo程序验证GPU是否正常,代码如下:
import pytorch
print(torch.__version__)
t=torch.ones(1,1,2,2)
t=torch.cuda()
发现报错:
RuntimeError: CUDA error: no kernel image is available for execution on the device
查明是torch的版本和cuda的版本不匹配,升级torch可以解决这个问题。
升级命令可以使用如下网页查询:
https:/
原创
2021-06-10 15:31:26 ·
822 阅读 ·
0 评论