安装pytorch-gpu时会默认安装cpu版本
安装pytorch-gpu时会默认安装cpu版本
安装pytorch-gpu版本时会默认安装cpu版本
使用官网中给出的指令安装pytorch-cuda版本
,但是安装完成后发现依旧为cpu版本
。
原因
在安装pytorch时会默认安装一个名为cpuonly的库,这使pytorch
以及torchvision
的版本都默认为cpu
版,即便你在安装时给出cudatoolkit版本也无效。
解决
1、在完成所有安装后,发现为cpu版本,先检查是否存在cpuonly库,在激活虚拟环境的命令行输入:
conda list
2、并不是每个人都会有cpuonly,例如本小白库中并没有cpuonly,但是pytorch版本仍为cpu
3、出现这种i情况,只需先将cpuonly安装,再卸载即可,命令行输入(若存在cpuonly库,直接跳转至第四步)
conda install cpuonly
4、接下来再执行
conda uninstall cpuonly
5、本小白执行结果如下:
6、可以看到cpu版本会转为gpu版本,输入y确认即可。
7、安装完成后重新检查:
python
import torch
print(torch.cuda.is_available())
8、返回True,即为gpu版本