cmd依次输入
python
import torch
print(torch.cuda.is_available())
会显示false
网上理由:
1.CUDA 版本不对这个可以cmd输入nvcc -V ,查看代码调用的CUDA是哪个版本
(网上说输入NVIDIA-smi会CUDA显示版本,但是它显示的CUDA version其实是你可以安装的最高版本的CUDA,而不是你会调用的CUDA)
2.torch版本不对cmd依次输入:
python
import torch
print(torch.__version__)
会显示你的pytorch版本
在anaconda prompt输入 conda list
可以发现pytorch是CPU还是GPU版,应该下载GPU版
pip uninstall torch后
去官网Previous PyTorch Versions | PyTorch
找命令下载,一步到位



