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

Question

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

在Google进行一些搜索之后，有人写了一篇关于找到仅CPU版本的PyTorch的文章，并使用它，在这里要小得多。。

但是，我对如何完成此操作非常迷惑，而该人员根本没有记录在案。任何建议表示赞赏，谢谢。

编辑：

为了更具体地说明我的问题，据我所知，我尝试安装torch by，其中包括require.txt，其中将torch列为依赖项。当前我有：火炬== 0.4.1。但是，这不起作用的大小。

我的问题是，您是否知道我可以在需求文件中写些什么来获得较小的仅用于cpu的割炬版本，或者，如果requirements.txt不适用于该版本，我会怎么做，以便获取CPU版本。

Answer 1

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

conda install pytorch torchvision cpuonly -c pytorch

并且，可以从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 的入门指南。

Answer 2

小智 13

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

`{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

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

每Pytorch网站，你可以安装pytorch-cpu使用

conda install pytorch-cpu torchvision-cpu -c pytorch

您可以从Anaconda云上的文件中看到，其大小在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 5

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

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

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