heroku / heroku/heroku-buildpack-python

Add warning when using `tensorflow` rather than `tensorflow-cpu`

Open
#1,343 0 comments 0 reactions 0 assignees View on GitHub
enhancement
Dominant language
Ruby
Stars
993
Forks
1.8k
Avg merge
10h 50m
Merged PRs (30d)
10

Description

The main `tensorflow` package wheel is ~500MB (!) which means anyone using it is likely to go over the Heroku slug limit (that limit is compressed slug size, however there will be other packages + the Python runtime + many of the compiled files won't be that compressible).

eg as seen in:
https://heroku.support/1140002
https://heroku.support/1529834

It turns out tensorflow has a CPU-only package available thats half the size. Given that Heroku does not offer GPUs, it seems worthwhile to warn if the larger CPU+GPU package is being used instead of the CPU one.

See:
https://github.com/tensorflow/tensorflow#install
https://pypi.org/project/tensorflow/#files
https://pypi.org/project/tensorflow-cpu/#files

GUS-W-19247170.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.