1.问题:
2.解决:
(1)命令行输入:
NVIDIA-smi
查看cuda的版本号和电脑驱动版本号是否匹配:
参考对比是否匹配:
(2)巨坑!!!!:检查自己安装的torch是cpu版本还是GPU版本,如果是CPU版本,需要卸载重装:
检查方法:
打开cmd输入:
python
import torch
print(torch.__version__)
输出:(有“+cpu”则说明是cpu版本)
卸载:有的电脑在后续安装新版本会自动检测卸载,如果不会的,需要找到路径(路径根据每个人安装的anaconda路径而异)
C:\Users\ASCV\anaconda3\Lib\site-packages
找到带torch字样的文件夹都删除
# ------------22.1.28补充------------------------------------
可能需要找到所有site-packages下删除带torch字样的文件夹(删除之间检查是否已正确删除要使用虚拟环境下的torch,可以进入python解释器,import torch看是否报错:ModuleNotFoundError: No module named 'torch',如果是,则表示删除成功))
#----------------------------------------------------------------
(3)安装torch:
进入官网:(可能需vpn)
https://pytorch.org/get-started/locally/#no-cuda-1
根据自己的cuda和电脑版本进行选择,得到下载指令,复制到cmd运行:
如果有vpn可能会下载的快一些
下载完成:
检验结果,cmd输入:
python
import torch
torch.cuda.is_available()
3.解决结果:
返回true