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

时间:2018-08-07 15:51:00

标签: heroku pytorch

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

在Google上进行一些搜索之后,有人写了一篇关于找到仅cpu版本的PyTorch,并使用它的版本,该版本要小得多here

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

编辑:

为了更具体地说明我的问题,据我所知,我尝试安装了割炬,其中包括一个require.txt,其中将割炬列为依赖项。当前我有:火炬== 0.4.1。但是,这不符合bc的大小。

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

6 个答案:

答案 0 :(得分:5)

在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-cpunumpyscipy,而您使用的是Python 3.6,则requirements.txt如下所示:

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

答案 1 :(得分:4)

如果要使用pip3下载pytorch-cpu版本(在Linux和Win上),请在2020年使用以下命令:

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

答案 2 :(得分:1)

我收到来自 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

答案 3 :(得分:0)

自PyTorch 1.3起,PyTorch更改了其API。为了仅安装CPU版本,请使用

conda install pytorch torchvision cpuonly -c pytorch

然后,可以从https://download.pytorch.org/whl/torch_stable.html下载相应的转轮文件,并可以使用pip安装相应的转轮文件,或者使用与以下类似的命令来匹配所需的pytorch和torchvision版本

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

查看PyTorch的{​​{3}}指南。

答案 4 :(得分:0)

如果要使用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版本)

答案 5 :(得分:-1)

正确的答案是在Pytorch网站上进行搜索,网址为https://pytorch.org/get-started/previous-versions/

它们具有所有以前版本的完整列表,以及每个版本使用的pip语法(包括仅CPU版本)。