当前位置: 首页 > news >正文

linxu下安装纯cpu模式的pytorch以及国内源安装方法

背景

最近发现制作的镜像比较大,看了下主要是python挺大的,看了下依赖里面有很多nvidia相关的依赖,但是实际上我并没有nvidia的gpu,通过pipdeptree工具看到这些包是pytorch引入的,所以就想只安装cpu版本的pytorch来减小镜像size。

pipdeptree可以通过下面的命令安装

pip install pipdeptree

在网络搜索了一圈,大部分的内容感觉都挺有误导性的,所以这里自己整理了一下,首先这里要搞清楚你要安装的平台,不同的平台默认版本是不一样的,这个是误导我最大的地方,因为很多内容都说默认就是cpu版本的,所以这里我整理了一下:

  1. windows 默认是cpu版本,cuda版本需要单独指出
  2. mac 默认是cpu版本,且只有cpu版本
  3. linux 默认是cuda版本的,cpu需要单独指出

下面主要说linux上面的cpu版本安装。

方法

官方源

这个方法比较直接,只取去pytorch官方的getstarted页面,选择要安装的pytorch的版本、操作系统,工具以及platform(cuda、cpu等),就会自动给你生成好命令供你安装:

pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu

阿里源

阿里源单独放置了cpu版本的pytorch,可以通过下面的命令来安装。

pip install  torch==2.8.0+cpu -i https://pypi.tuna.tsinghua.edu.cn/simple -f https://mirrors.aliyun.com/pytorch-wheels/cpu

其他源我没有找到安装方式,不知道是没有还是我没有找到,如果读者朋友有其他安装方式,欢迎在评论区指出来。

http://www.dtcms.com/a/537499.html

相关文章:

  • C++ 常见代码异味(Code Smells)
  • 手写线程池第2弹:并发与并行深度解析:从CPU原理到高并发系统设计的核心技术
  • 国产三维CAD实现复杂实体快速转换钣金 | 中望3D 2026亮点速递(9)
  • 免费网站建设专业的公司软件设计师教程
  • 关于企业的网站wordpress建站seo
  • Java数据结构:Stack(栈)Queue(队列)
  • Python中的列表推导式、字典推导式和集合推导式的性能和应用场景?
  • Spring全家桶面试题, 只补充细节版本
  • 做网站可以挣钱吗微信公众平台网页版登录
  • 第2章-类加载子系统
  • 网站左侧悬浮导航芜湖学校网站建设电话
  • PyTorch2 Python深度学习 - 简介以及入门
  • 定制版网站建设费用湘潭网站建设湘潭振企专业
  • 自己做的小网站如何发布网络营销案例范文
  • 单体架构中的事件驱动架构:Java应用程序的渐进式重构
  • 成都网站开发的公司吉安县规划建设局网站
  • 有了域名怎么建设网站淘宝客网站是怎么做的
  • 铁岭做网站大学网页设计与制作教程
  • 工商工事上哪个网站做西安网站制作公司排名
  • 机关单位网站建设申请公司简介模板图片
  • 【Delphi】操纵EXE文件中的主图标(MAINICON)
  • 彭州建设网站陕西省建设执业资格注册中心网站
  • 济南企业上云网站建设如何让网站自适应手机
  • nginx基础入门篇-nginx部署-Yum
  • Leetcode每日一练--43
  • Nacos 配置中心:动态配置管理
  • cpp / c++零基础两周速成教学
  • 免费微场景制作网站深圳外贸公司有哪些公司
  • Selenium定位元素的方法css和xpath的区别
  • 什么网站可以自学ps做贵宾卡互联网营销师