Specify TF CPU as an extra during installation

uber · Jul 20, 2018 · 86407ef · 86407ef
1 parent 04e6ee4
commit 86407ef
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.travis.yml b/.travis.yml
@@ -6,7 +6,8 @@ python:
 
 install:
   # This will use requirements from setup.py and install them in the tavis's virtual environment
-  - pip install -e .
+  # [tf] chooses to depend on cpu version of tensorflow (alternatively, could do [tf_gpu])
+  - pip install -e .[tf]
 
 script:
   # Run unit tests