Fix scratchgan requirements and update README.

PiperOrigin-RevId: 332257241
kabajau · Sep 18, 2020 · fd75024 · fd75024
1 parent 3043408
commit fd75024
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/scratchgan/README.md b/scratchgan/README.md
@@ -52,6 +52,11 @@ Create and activate a virtual environment if needed:
     virtualenv scratchgan-venv
     source scratchgan-venv/bin/activate
 
+
+Note: the code uses python 2. This might mean that to create the virtual env
+you have to pass a path to the python binary to be used, as follows:
+`virtualenv -p /usr/bin/python2.7 scratchgan-venv`.
+
 Install requirements:
 
     pip install -r scratchgan/requirements.txt

diff --git a/scratchgan/requirements.txt b/scratchgan/requirements.txt
@@ -1,8 +1,9 @@
 absl-py==0.7.1
 dm-sonnet==1.34
 numpy==1.16.4
-tensorflow==1.14
+tensorflow==1.15
 tensorflow-probability==0.7.0
 tensorflow-gan == 1.0.0.dev0
 tensorflow-hub == 0.6.0
 tensorflow-io == 0.8.0
+tensorflow-gpu==1.15