运维系列&AI系列&Lenovo-G双系统系列(前传-装之前搜索的):安装pytorch-gpu时会默认安装cpu版本
安装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版本
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· 上周热点回顾(2.24-3.2)