Fix TensorFlow requirement

IBM / tensorflow-large-model-support

Large Model Support in Tensorflow

Apache License 2.0

202 stars 38 forks source link

Fix TensorFlow requirement #15

Closed smatzek closed 5 years ago

smatzek commented 5 years ago

Fix the TensorFlow requirement in setup.py to be "tensorflow-gpu" rather than "tensorflow". This module is only intended for use with GPU enabled TensorFlow.

Fixes: #14