运维系列&AI系列&Lenovo-G双系统系列(前传-装之前搜索的):安装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版本

在这里插入图片描述







勇往无前猪猪侠

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

posted @ 2024-08-25 17:30  坦笑&&life  阅读(26)  评论(0编辑  收藏  举报  来源