matrix

.travis.yml

 language: python
 
-python:
-  - "3.6"
-  - "3.7"
-  - "3.8"
-
 sudo: enabled
 
 compiler:
   - gcc
 
-matrix:
+jobs:
   include:
-    - name: cu92
-      env:
-        - FORCE_CUDA=1
-        - CUDA=9.2.148-1
-        - CUDA_SHORT=9.2
-        - UBUNTU_VERSION=ubuntu1604
-        - CUBLAS=cuda-cublas-dev-${CUDA_SHORT/./-}
-        - TORCH=1.4.0+cu92
-    - name: cu100
+    - os: linux
+      python:
+        - '3.6'
+        - '3.7'
       env:
-        - FORCE_CUDA=1
-        - CUDA=10.0.130-1
-        - CUDA_SHORT=10.0
-        - UBUNTU_VERSION=ubuntu1804
-        - CUBLAS=cuda-cublas-dev-${CUDA_SHORT/./-}
-        - TORCH=1.4.0+cu100
-    - name: cu101
-      env:
-        - FORCE_CUDA=1
-        - CUDA=10.1.105-1
-        - CUDA_SHORT=10.1
-        - UBUNTU_VERSION=ubuntu1804
-        - CUBLAS=libcublas-dev
-        - TORCH=1.4.0
-
-before_install:
-  - INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
-  - wget http://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/${INSTALLER}
-  - sudo dpkg -i ${INSTALLER}
-  - wget https://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/7fa2af80.pub
-  - sudo apt-key add 7fa2af80.pub
-  - sudo apt update -qq
-  - sudo apt install -y cuda-core-${CUDA_SHORT/./-} cuda-cudart-dev-${CUDA_SHORT/./-} ${CUBLAS} cuda-cusparse-dev-${CUDA_SHORT/./-}
-  - sudo apt clean
-  - CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
-  - LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
-  - PATH=${CUDA_HOME}/bin:${PATH}
+        - FORCE_CUDA=1 CUDA=9.2.148-1 CUDA_SHORT=9.2 UBUNTU_VERSION=ubuntu1604 CUBLAS=cuda-cublas-dev-9-2 TORCH=1.4.0+cu92
+        - FORCE_CUDA=1 CUDA=10.0.130-1 CUDA_SHORT=10.0 UBUNTU_VERSION=ubuntu1804 CUBLAS=cuda-cublas-dev-10-0 TORCH=1.4.0+cu100
+        - FORCE_CUDA=1 CUDA=10.1.105-1 CUDA_SHORT=10.1 UBUNTU_VERSION=ubuntu1804 CUBLAS=libcublas-dev TORCH=1.4.0
+      before_install:
+        - INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
+        - wget http://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/${INSTALLER}
+        - sudo dpkg -i ${INSTALLER}
+        - wget https://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/7fa2af80.pub
+        - sudo apt-key add 7fa2af80.pub
+        - sudo apt update -qq
+        - sudo apt install -y cuda-core-${CUDA_SHORT/./-} cuda-cudart-dev-${CUDA_SHORT/./-} ${CUBLAS} cuda-cusparse-dev-${CUDA_SHORT/./-}
+        - sudo apt clean
+        - CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
+        - LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+        - PATH=${CUDA_HOME}/bin:${PATH}
 
 install:
   - pip install numpy
@@ -56,7 +34,6 @@ install:
   - pip install flake8
 
 script:
-  - nvcc --version
   - python -c "import torch; print(torch.__version__)"
   - flake8 .
   - python setup.py install