运维系列&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 @   坦笑&&life  阅读(47)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· 上周热点回顾(2.24-3.2)
点击右上角即可分享
微信分享提示