在哪里可以获得仅 CPU 版本的 PyTorch？

Question

我正在尝试使用 Flask + PyTorch 运行一个基本应用程序，并将其托管在 Heroku 上。但是，我遇到的问题是免费版本的最大 slug 大小为 500mb，而 PyTorch 本身约为 500mb。

经过一些谷歌搜索后，有人写了关于找到 PyTorch 的纯 cpu 版本并使用它，它要小得多here。

但是，我对这是如何完成的感到非常困惑，并且该人根本没有记录这一点。任何建议表示赞赏，谢谢。

编辑：

为了更具体地说明我的问题，我尝试安装 torch（据我所知），包括将 torch 列为依赖项的requirements.txt。目前我有：torch==0.4.1。然而，这对于大小来说不起作用。

我的问题是，你知道我可以在需求文件中写什么来获得较小的仅CPU版本的torch，或者，如果requirements.txt对此不起作用，我会做什么，获取CPU版本。

Answer 1

根据 Pytorch 网站，您可以使用

 安装 

pytorch-cpu

conda install pytorch-cpu torchvision-cpu -c pytorch

您可以从 Anaconda cloud 上的文件中看到，大小在 26 到 56MB 之间变化，具体取决于您要安装的操作系统。

您可以从

http://download.pytorch.org/whl/cpu/

获得轮子。 轮子有87MB。

您可以通过将轮子的链接放入

requirements.txt

文件中来设置安装。如果您在 Heroku 上使用 Python 3.6：

http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl

否则，对于 Python 2.7：

http://download.pytorch.org/whl/cpu/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl

例如，如果您的要求是

pytorch-cpu

、

numpy

和

scipy

并且您使用的是 Python 3.6，则

requirements.txt

将如下所示：

http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
numpy
scipy

Answer 2

从 PyTorch 1.3 开始，PyTorch 更改了其 API。为了仅安装 CPU 版本，请使用

conda install pytorch torchvision cpuonly -c pytorch

并且，可以从 https://download.pytorch.org/whl/torch_stable.html 下载相应的 Wheel 文件，并可以使用

pip

进行安装，或者使用与您想要的 pytorch 相对应的类似于以下的命令和火炬视觉版本

在 Linux 上：

pip3 install torch==1.9.0+cpu torchvision==0.10.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

在 Windows / Mac 上：

pip3 install torch torchvision

查看 PyTorch 的入门指南。

Answer 3

2020年，如果您想使用pip3下载pytorch-cpu版本（在Linux和Win上），请使用以下命令：

pip3 install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

Answer 4

我从 torch 稳定版本列表中收到每个版本的错误。喜欢

`{specific_version} 不是此平台支持的滚轮

尝试将其放入您的

// requirements.txt

-f https://download.pytorch.org/whl/torch_stable.html
torch==1.8.1+cpu 
torchvision==0.9.1+cpu
fastai>=2.3.1
ipywidgets
voila

Answer 5

在 Heroku 的应用程序平台遇到同样的问题后，来到这个问题——slug 大小well超过 500MB 限制。官方PyTorch“入门”页面当前的说明如下：

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

这是 Linux 安装，使用最新的（1.13.1）稳定版本，Python 和 pip。注意：语法因系统、包管理器、语言和首选构建而异。 （见下文）

Answer 6

您可以使用 pip 直接从 pytorch.org 网站下载最新的仅 CPU 的 pytorch 轮：

pip install torch --extra-index-url https://download.pytorch.org/whl/cpu

Answer 7

问题在于库的大小，当您在本地使用应用程序时，您可以使用GPU资源，因为您不会在服务器上使用它，请在requirements.txt中使用以下代码：

--find-links https://download.pytorch.org/whl/torch_stable.html
torch==1.11.0+cpu
--find-links https://download.pytorch.org/whl/torch_stable.html
torchvision==0.12.0+cpu

Answer 8

如果您想使用requirements.txt安装稳定的pytorch=1.4.0 cpu版本，则指定直接下载http链接... 这样pip就会直接下载并安装...

http://download.pytorch.org/whl/cpu/torch-1.4.0%2Bcpu-cp37-cp37m-linux_x86_64.whl 

或者如果使用终端或cmd

torch==1.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

更多版本请访问 https://download.pytorch.org/whl/torch_stable.html并根据您的要求选择版本（windows、linux、mac版本都可以在链接中看到）

Answer 9

如果您尝试仅使用

requirements.txt

解决此问题，请将以下内容放入您的

requirements.txt

中：

--index-url https://download.pytorch.org/whl/cpu
--extra-index-url https://pypi.org/simple
torch==2.2.1
Flask==3.0.2

基本上，它告诉 pip 首先尝试查找

https://download.pytorch.org/whl/cpu

中的每个包，这将为您提供仅 CPU 版本的 torch。对于每个其他包，它将回退到默认索引。

Answer 10

正确的答案是在 Pytorch 网站上搜索它，这里：https://pytorch.org/get-started/previous-versions/

他们有所有以前版本的完整列表，以及每个版本使用的 pip 语法（包括仅 CPU 版本）。