linxu下安装纯cpu模式的pytorch以及国内源安装方法
背景
最近发现制作的镜像比较大,看了下主要是python挺大的,看了下依赖里面有很多nvidia相关的依赖,但是实际上我并没有nvidia的gpu,通过pipdeptree工具看到这些包是pytorch引入的,所以就想只安装cpu版本的pytorch来减小镜像size。
pipdeptree可以通过下面的命令安装
pip install pipdeptree
在网络搜索了一圈,大部分的内容感觉都挺有误导性的,所以这里自己整理了一下,首先这里要搞清楚你要安装的平台,不同的平台默认版本是不一样的,这个是误导我最大的地方,因为很多内容都说默认就是cpu版本的,所以这里我整理了一下:
- windows 默认是cpu版本,cuda版本需要单独指出
- mac 默认是cpu版本,且只有cpu版本
- 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
其他源我没有找到安装方式,不知道是没有还是我没有找到,如果读者朋友有其他安装方式,欢迎在评论区指出来。
