我试图得到一个基本的应用程序运行与 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版本。
8条答案
按热度按时间icomxhvb1#
根据Pytorch网站,您可以使用
您可以从Anaconda cloud上的文件中看到,大小在26到56MB之间变化,具体取决于您要安装它的操作系统。
你可以从
http://download.pytorch.org/whl/cpu/
获得轮子。轮子是87MB。你可以通过在
requirements.txt
文件中放置指向wheel的链接来设置安装。否则,对于Python 2.7:
例如,如果您的需求是
pytorch-cpu
、numpy
和scipy
,并且您使用的是Python 3.6,则requirements.txt
将如下所示:baubqpgj2#
从PyTorch 1.3开始,PyTorch已经改变了它的API。为了只安装CPU版本,使用
而且,可以从https://download.pytorch.org/whl/torch_stable.html下载相应的wheel文件,并可以使用
pip
或使用类似于以下命令的命令安装wheel文件,这些命令对应于您要使用的pytorch和torchvision版本在Linux上:
在Windows / Mac上:
查看PyTorch的getting started指南。
az31mfrm3#
在2020中,如果您想下载pytorch-cpu版本的pip 3(在Linux和Win上),请使用以下命令:
dwthyt8l4#
我从list of torch stable versions得到每个版本的错误。like
'{specific_version}不是此平台上支持的轮
试着把这个放进你的requirements.txt
vatpfxk55#
如果你想使用requirements.txt安装稳定的pytorch=1.4.0 cpu版本,那么指定直接下载http链接...这样pip将直接下载并安装...
或者,如果使用终端或cmd
更多版本,请访问https://download.pytorch.org/whl/torch_stable.html并根据您的要求选择版本(windows、linux、mac版本均可在链接中看到)
z18hc3ub6#
问题是库的大小,当你在本地使用应用程序时,你可以使用GPU资源,因为你不会在服务器上使用它,使用requirements.txt中的以下代码:
bakd9h0s7#
您可以使用pip直接从www.example.com网站下载最新的纯CPU pytorch wheelpytorch.org:
gblwokeq8#
正确答案是在Pytorch网站上搜索,网址为:https://pytorch.org/get-started/previous-versions/
他们有一个完整的列表,列出了所有以前的版本,以及用于每个版本的pip语法(包括纯CPU版本)。