我正在尝试使用Flask + PyTorch运行一个基本应用程序,并将其托管在Heroku上。但是,我遇到的问题是,免费版本的最大子弹大小为500mb,而PyTorch本身约为500mb。
在Google上进行一些搜索之后,有人写了一篇关于找到仅cpu版本的PyTorch,并使用它的版本,该版本要小得多here。
但是,我对如何完成此操作非常迷惑,而该人员根本没有对此进行记录。任何建议表示赞赏,谢谢。
编辑:
为了更具体地说明我的问题,据我所知,我尝试安装了割炬,其中包括一个require.txt,其中将割炬列为依赖项。当前我有:火炬== 0.4.1。但是,这不符合bc的大小。
我的问题是,您知道我可以在需求文件中写些什么来获得较小的仅用于cpu的割炬版本,或者,如果requirements.txt不适用于该版本,我会怎么做而是获取cpu版本。
答案 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-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
答案 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版本)。