update cuda 10.1

.travis.yml

@@ -13,6 +13,12 @@ matrix:
         - CUDA=9.2.148-1
         - CUDA_SHORT=9.2
         - UBUNTU_VERSION=ubuntu1604
+    - name: Linux/CUDA 10.1
+      env:
+        - FORCE_CUDA=1
+        - CUDA=10.1.105-1
+        - CUDA_SHORT=10.1
+        - UBUNTU_VERSION=ubuntu1804
 
 before_install:
   - INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
@@ -21,7 +27,7 @@ before_install:
   - 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-${CUDA_SHORT/./-} 
+  - sudo apt install -y cuda-${CUDA_SHORT/./-}
   - sudo apt clean
   - CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
   - LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
@@ -29,7 +35,7 @@ before_install:
 
 install:
   - pip install numpy
-  - pip install torch==1.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
+  - pip install torch==1.4.0+cu${CUDA_SHORT/./} -f https://download.pytorch.org/whl/torch_stable.html
 
 script:
   - nvcc --version