【bug】RuntimeError: CUDA error: no kernel image is available for execution on the device 解决方案
使用demo程序验证GPU是否正常,代码如下:import pytorchprint(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 ·
854 阅读 ·
0 评论