diff --git a/.travis.yml b/.travis.yml
index f9b54ba26..3bbbe4422 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -48,7 +48,9 @@ matrix:
         - #mv test test.orig
         - wget https://www.mkroeker.com/pytorch.tbz
         - tar jxf pytorch.tbz
-        - cd pytorch/test
+        - cd pytorch
+        - python3 setup.py install
+        - cd test
         - sed -i -e "s/@onlyCUDA//g" -e "s/@largeTensorTest('12GB')//g" test_nn.py
         - python3 test_nn.py TestNNDeviceTypeCPU.test_conv_large_cpu
         - cp ../../libopenblas.so $HOME/conda/lib