heroku 我在哪里可以得到一个纯CPU版本的PyTorch?

yi0zb3m4  于 2022-12-13  发布在  其他
关注(0)|答案(8)|浏览(169)

我试图得到一个基本的应用程序运行与 flask + PyTorch,并在Heroku上托管它。然而,我遇到的问题,最大的子弹大小是500 mb的免费版本,和PyTorch本身是~ 500 mb。
经过一些谷歌搜索,有人写道,找到一个cpu版本的PyTorch,并使用它,这是小得多的here
然而,我对如何做到这一点相当困惑,而且这个人根本没有记录这一点。任何建议都是感激的,谢谢。
编辑:
为了更具体地说明我的问题,我尝试通过(据我所知)安装torch,其中包括一个requirements.txt,它将torch列为依赖项。 Torch ==0.4.1。然而,这不适用于bc的大小。
我的问题是,你知道我可以在需求文件中写些什么来获得更小的只有cpu的torch版本吗?或者,如果requirements.txt不起作用,我会做些什么来获得cpu版本。

icomxhvb

icomxhvb1#

根据Pytorch网站,您可以使用

conda install pytorch-cpu torchvision-cpu -c pytorch

您可以从Anaconda cloud上的文件中看到,大小在26到56MB之间变化,具体取决于您要安装它的操作系统。
你可以从http://download.pytorch.org/whl/cpu/获得轮子。轮子是87MB。
你可以通过在requirements.txt文件中放置指向wheel的链接来设置安装。

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
baubqpgj

baubqpgj2#

从PyTorch 1.3开始,PyTorch已经改变了它的API。为了只安装CPU版本,使用

conda install pytorch torchvision cpuonly -c pytorch

而且,可以从https://download.pytorch.org/whl/torch_stable.html下载相应的wheel文件,并可以使用pip或使用类似于以下命令的命令安装wheel文件,这些命令对应于您要使用的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指南。

az31mfrm

az31mfrm3#

在2020中,如果您想下载pytorch-cpu版本的pip 3(在Linux和Win上),请使用以下命令:

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

dwthyt8l4#

我从list of torch stable versions得到每个版本的错误。like
'{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
vatpfxk5

vatpfxk55#

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

z18hc3ub

z18hc3ub6#

问题是库的大小,当你在本地使用应用程序时,你可以使用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
bakd9h0s

bakd9h0s7#

您可以使用pip直接从www.example.com网站下载最新的纯CPU pytorch wheelpytorch.org:

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

gblwokeq8#

正确答案是在Pytorch网站上搜索,网址为:https://pytorch.org/get-started/previous-versions/
他们有一个完整的列表,列出了所有以前的版本,以及用于每个版本的pip语法(包括纯CPU版本)。

相关问题