install process for tensorflow-gpu (#49)

Author: bfineran

README.md

@@ -321,7 +321,15 @@ TensorFlow V1:
 
 ```bash
 pip install sparseml[tf_v1]
-```
+``` 
+
+TensorFlow V1 with GPU operations enabled:
+```bash
+pip install sparseml[tf_v1_gpu]
+``` 
+Depending on your device and CUDA version, you may need to install additional
+dependencies for using TensorFlow V1 with GPU operations.  You can find these
+steps [here](https://www.tensorflow.org/install/gpu#older_versions_of_tensorflow).
 
 ## Resources and Learning More
 

setup.py

@@ -39,6 +39,11 @@
 _pytorch_deps = ["torch>=1.1.0", "tensorboard>=1.0", "tensorboardX>=1.0"]
 _pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0"]
 _tensorflow_v1_deps = ["tensorflow<2.0.0", "tensorboard<2.0.0", "tf2onnx>=1.0.0,<1.6"]
+_tensorflow_v1_gpu_deps = [
+    "tensorflow-gpu<2.0.0",
+    "tensorboard<2.0.0",
+    "tf2onnx>=1.0.0,<1.6",
+]
 _keras_deps = ["tensorflow~=2.2", "keras2onnx>=1.0.0"]
 
 _dev_deps = [
@@ -74,6 +79,7 @@ def _setup_extras() -> Dict:
         "torch": _pytorch_deps,
         "torchvision": _pytorch_vision_deps,
         "tf_v1": _tensorflow_v1_deps,
+        "tf_v1_gpu": _tensorflow_v1_gpu_deps,
         "tf_keras": _keras_deps,
     }