运维系列&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版本

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

坦笑&&life

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值