require tensorflow-gpu instead of tensorflow

This commit is contained in:
Robert Sachunsky 2020-11-23 10:07:58 +01:00 committed by GitHub
parent 8c74d717fc
commit a046699b0b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -4,4 +4,4 @@ opencv-python-headless
ocrd >= 2.18.0 ocrd >= 2.18.0
keras >= 2.3.1, < 2.4 keras >= 2.3.1, < 2.4
h5py < 3 h5py < 3
tensorflow >= 1.15, < 1.16 tensorflow-gpu >= 1.15, < 1.16