安装pytorch-gpu时会默认安装cpu版本

Posted 2023-01-06 勇往无前猪猪侠

目录

• 安装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版本