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

【问题标题】：Where do I get a CPU-only version of PyTorch?我在哪里可以获得仅 CPU 版本的 PyTorch？
【发布时间】：2019-01-14 18:51:36
【问题描述】：

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

在谷歌搜索后，有人写到找到一个仅 cpu 版本的 PyTorch，并使用它，它要小得多here。

但是，我不知道这是如何完成的，而且这个人根本没有记录这一点。任何建议表示赞赏，谢谢。

编辑：

为了更具体地说明我的问题，我尝试安装torch by（据我所知），包括一个将torch列为依赖项的requirements.txt。目前我有：手电筒==0.4.1。但是，这在大小上不起作用。

我的问题是，你知道我可以在需求文件中写什么来获得更小的仅 cpu 版本的火炬，或者，如果 requirements.txt 不适合这个，我会怎么做相反，要获取 cpu 版本。

【问题讨论】：

要求我们推荐或查找书籍、工具、软件库、教程或其他场外资源的问题对于 Stack Overflow 来说是无关紧要的，因为它们往往会吸引固执己见的答案和垃圾邮件。相反，请描述问题以及迄今为止为解决该问题所做的工作。
抱歉——这不是我的本意。我并不是想寻求建议，我更想问的是如何正确使用我没有经验的 heroku。我会更新我的问题。

【解决方案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

【讨论】：

感谢您的回复。我知道如何在我自己的本地机器上安装它，但你能更具体地说明如何在 Heroku 上执行此操作吗？我是否以特定方式编辑 requirements.txt，或者如果我直接获取 whl 文件，我如何让 Heroku 为我的应用安装它？
更新了我的答案，你必须在requirements.txt中包含指向轮子的链接
mkl 是用pytorch cpu whl 打包的吗？
能否请您告诉如何在不使用 conda 的情况下安装 pytorch-cpu 和 torchvision-cpu。我已经创建了一个虚拟环境，我想在那里安装它。（CPU 机器，Ubuntu 16.04，不支持 CUDA）。谢谢。
在 Python 3.6 pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl 和 pip3 install torchvision 上试试这个

【解决方案2】：

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

conda install pytorch torchvision cpuonly -c pytorch

并且，对应的wheel文件可以从https://download.pytorch.org/whl/torch_stable.html下载，可以使用pip安装，或者使用类似下面的命令对应你想要的pytorch和torchvision版本

在 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 的 getting started 指南。

【讨论】：

【解决方案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

【讨论】：

【解决方案4】：

我收到来自list of torch stable versions 的每个版本的错误。喜欢

`{specific_version} 在此平台上不支持

试着把这个放到你的requirements.txt

// 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

【讨论】：

【解决方案5】：

如果你想使用 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版本都可以在链接中看到）

【讨论】：

【解决方案6】：

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

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

【讨论】：