python - Heroku:安装 Pytorch 后 slug 尺寸太大

Question

我一直收到来自 Heroku 的 slug size too large 警告(Compiled slug size: 789.8M is too large (max is 500M))，我不知道为什么，因为我的模型大小(下面的 cnn.pth)相当小，我的文件目录总共只有 1.1mb:screenshot of directory.

似乎大小增加是由运行 pipenv install torch 引起的，因为 slug 大小在安装 torch 之前为 89.1mb，安装之后为 798.8mb。

我的 Pipfile 目前安装了这些包:

[packages]
flask = "*"
flask-sqlalchemy = "*"
psycopg2 = "*"
psycopg2-binary = "*"
requests = "*"
numpy = "*"
gunicorn = "*"
pillow = "*"
torch = "*"

有解决办法吗？

编辑:我正在运行 Mac OSX 10.10.5，使用 Flask 和 pipenv。

Answer 1

您正在安装的 pytorch 软件包同时支持 cpu 和 gpu，因此体积较大。看来您使用的是 heroku 的免费版本，并且只需要 cpu 支持。解决方案是只为 cpu 安装 pytorch 包，即

在requirements.txt中，写上你感兴趣的pytorch(cpu)版本对应的wheel文件路径，可以找到list of wheel files ，可以使用 pip 安装。比如PyTorch 1.3.1, torchvision 0.4.2, Python 3.7, Linux，可以分别为pytorch和torchvision写如下:

https://download.pytorch.org/whl/cpu/torch-1.3.1%2Bcpu-cp37-cp37m-linux_x86_64.whl
https://download.pytorch.org/whl/cpu/torchvision-0.4.2%2Bcpu-cp37-cp37m-linux_x86_64.whl

上面会下载torch-1.3.1+cpu-cp37-cp37m-linux_x86_64.whl (107MB) torchvision-0.4.2+cpu-cp37-cp37m-linux_x86_64.whl (13MB) 分别。