separate out tensorflow install

jankim · Feb 20, 2019 · 825aa3d · 825aa3d
1 parent fc0ee6d
commit 825aa3d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -13,7 +13,18 @@ Download the model data
 sh download_model.sh 117M
 ```
 
-Install python packages:
+The remaining steps can optionally be done in a virtual environment using tools such as `virtualenv` or `conda`.
+
+Install tensorflow 1.12 (with GPU support, if you have a GPU and want everything to run faster)
+```
+pip3 install tensorflow==1.12.0
+```
+or
+```
+pip3 install tensorflow-gpu==1.12.0
+```
+
+Install other python packages:
 ```
 pip3 install -r requirements.txt
 ```

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,2 @@
 fire>=0.1.3
-tensorflow>=1.12
 regex==2017.4.5