减少机器学习的Heroku Slug大小(Python，PyTorch，Fastai)

xghobddn 于 2022-12-04 发布在 Python

关注(0)|答案(3)|浏览(122)

我试图部署一个简单的maching学习应用程序到heroku，但我一直超过500MB的大小要求，它看起来像在最后我来了约1GB。这似乎是来自PyTorch约700MB。

Collecting torch>=1.0.0
   Downloading torch-1.6.0-cp36-cp36m-manylinux1_x86_64.whl (748.8 MB)

我的requirements.txt文件看起来像

tensorboardX==1.6
opencv-python>=3.3.0.10
pillow>=6.2.1
flask
scikit-image
gunicorn
pandas

我收到的错误信息表明我超过了鼻涕虫的大小限制。
我怎么能只安装CPU版本的PyTorch来降低弹头的大小呢？

heroku

来源：https://stackoverflow.com/questions/63552330/reduce-heroku-slug-size-for-machine-learning-python-pytorch-fastai

3条答案

按热度按时间

sgtfey8w1#

尝试将以下行添加到requirements.txt

-f https://download.pytorch.org/whl/torch_stable.html
torch==1.8.1+cpu
torchvision==0.9.1+cpu
fastai
voila
ipywidgets

赞(0）回复(0）举报 2022-12-04

ecr0jaav2#

(Aug，2，2022）我找到的唯一解决方案是将requirements.txt保留为如下形式：
--查找链接https://download.pytorch.org/whl/torch_stable.html
焊炬==1.11.0+CPU
--查找链接https://download.pytorch.org/whl/torch_stable.html
Torch 视觉==0.12.0+CPU

赞(0）回复(0）举报 2022-12-04

gk7wooem3#

要安装PyTorch的CPU版本，可以在requirements.txt文件中指定cpuonly版本，如下所示：
torch==1.6.0+cpu
这将安装PyTorch的CPU版本，它的大小应该比GPU版本小得多。您也可以在requirements.txt文件中指定要安装的PyTorch的具体版本，在本例中是1.6.0。
一旦你更新了requirements.txt文件，你就可以运行pip install -r requirements.txt来安装所需的软件包。这应该会安装PyTorch的CPU版本，并减少应用的整体大小。

赞(0）回复(0）举报 2022-12-04

我来回答

减少机器学习的Heroku Slug大小(Python，PyTorch，Fastai)

3条答案

相关问题

热门标签

最新问答