安装GPU版本pytorch

 在conda虚拟环境中安装pytorch,可能最后检查能否使用gpu时会显示FALSE
image
 装好了cuda+cudnn+pytorch但是出现上述问题
 之前将conda源换成了清华的,安装pytorch不够智能,会识别成cpu版
 查看环境包如下


解决

法1.卸载cpuonly

查看conda list下是否存在cpuonly,若不存在,先手动安装

conda install cpuonly

然后卸载cpuonly

conda uninstall cppuonly

image
选择yes,之后他会自动重新下载pytorch gpu版本

法2.命令安装

进入pytorch官网,找到cuda的版本对应的命令,我的是11.4,这里选择11.3也是可以的


在命令行粘贴该命令

可以看到没有cpu字样了
等待安装完成
image
vans