当前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()