在conda虚拟环境中安装pytorch,可能最后检查能否使用gpu时会显示FALSE
装好了cuda+cudnn+pytorch但是出现上述问题
之前将conda源换成了清华的,安装pytorch不够智能,会识别成cpu版
查看环境包如下
解决
法1.卸载cpuonly
查看conda list下是否存在cpuonly,若不存在,先手动安装
conda install cpuonly
然后卸载cpuonly
conda uninstall cppuonly
选择yes,之后他会自动重新下载pytorch gpu版本
法2.命令安装
进入pytorch官网,找到cuda的版本对应的命令,我的是11.4,这里选择11.3也是可以的
在命令行粘贴该命令
可以看到没有cpu字样了
等待安装完成
vans