解决CUDA error: no kernel image is available for execution on the device 测试cuda是否可用


当前GPU的算力与当前版本的Pytorch依赖的CUDA算力不匹配(3080算力为8.6,而当前版本的pytorch依赖的CUDA算力仅支持3.7,5.0,6.0,7.0)

我的解决方法是重新到清华源网站上下载了pytorch,解决

链接是不同pytorch和gpu cuda cudnn版本匹配:

https://discuss.pytorch.org/t/gpu-compute-capability-support-for-each-pytorch-version/62434/7

测试cuda是否可用:

import torch
torch.cuda.is_available()
# 返回True 接着用下列代码进一步测试
torch.zeros(1).cuda()

相关