安装pytorch中始终使用的是cpu版本的pytorch 没有办法使用GPU运行代码
针对
import torch
torch.cuda.is_available()
输出
False
解决办法:
1. 参照了网上许多大佬的意见升级了NVIDIA,确保是430.0以上的版本
结果:没用 还是False
2. 重新安装pytorch 将版本对应cuda版本
结果:没用 还是False
3.从conda list中发现了cpuonly这个东西,并且自己的pytorch后面都是带有cpu的字样,于是果断卸载cpuonly。卸载后自动重装pytorch,对应的是GPU的pytorch。感觉行了!!!
看了眼重装完以后的list:
终于是GPU版本的了,再试验一次看看能不能成功:
终于成功了!
出现问题,可以参考通过conda list看里面的文件,从而进行修改