diff --git a/.travis.yml b/.travis.yml
index 8395be8dd80130904ea033beb954c8d4285aa18a..2b7fa740b496db71451694b5dfd47e0c2789ed16 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -26,6 +26,7 @@ before_script:
 script:
   - export NO_GPU_EXECUTION=1
   - pip install -e .
+  - pip install tensorflow torch
   - if [[ "$LINT" == "false" ]]; then python setup.py test; fi
   - if [[ "$LINT" == "true" ]]; then flake8 src;python setup.py doctest; fi
 after_success: