减少机器学习的 Heroku Slug 大小（Python、PyTorch、Fastai）答案

【问题标题】：Reduce Heroku Slug Size for Machine Learning (Python, PyTorch, Fastai)减少机器学习的 Heroku Slug 大小（Python、PyTorch、Fastai）
【发布时间】：2021-06-24 09:46:22
【问题描述】：

我正在尝试将一个简单的加工学习应用程序部署到 heroku，但我一直超过 500MB 的 slug 大小要求，看起来最终我达到了大约 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

我收到的错误消息表明我超过了 slug 大小限制。

我怎样才能只安装 CPU 版本的 PyTorch 来减小 slug 的大小？

【问题讨论】：

这能回答你的问题吗？ Where do I get a CPU-only version of PyTorch?

标签： python heroku pip

【解决方案1】：

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

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

【讨论】：