update

.travis.yml

@@ -9,6 +9,7 @@ matrix:
   include:
     - name: Linux/CUDA 9.2
       env:
+        - FORCE_CUDA=1
         - CUDA=9.2.148-1
         - CUDA_SHORT=9.2
         - UBUNTU_VERSION=ubuntu1604
@@ -28,7 +29,7 @@ before_install:
 
 install:
   - pip install numpy
-  - pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
+  - pip install torch==1.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
 
 script:
   - nvcc --version