Heroku：安装 Pytorch 后 slug 太大

Question

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

似乎大小增加是由运行 pipenv install torch 引起的，因为在安装 torch 之前 slug 大小为 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)。

Answer 2

1. 转到PyTorch Get Started page。选择稳定版，Linux，pip，python，cpu，可以看到：

pip install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

1. 将版本复制到您的requirements.txt。

torch==1.8.1+cpu 
torchvision==0.9.1+cpu

1. 并添加到requirements.txt 的开头这个：

-f https://download.pytorch.org/whl/torch_stable.html

之后，所有的包都将安装在 Heroku 上。

供参考： torch==1.8.1+cpu (169.1 MB); torchvision==0.9.1+cpu (13.3 MB)