解决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()