test cuda build

.travis.yml

-jobs:
+language: python
+
+sudo: enabled
+
+compiler:
+  - gcc
+
+matrix:
   include:
-    - os: linux
-      language: python
-      python: 3.7
-      addons:
-        apt:
-          sources:
-            - ubuntu-toolchain-r-test
-          packages:
-            - gcc-5
-            - g++-5
+    - name: CUDA 9.2
       env:
-        - CC=gcc-5
-        - CXX=g++-5
+        - CUDA=9.2.148-1
+        - CUDA_SHORT=9.2
+        - UBUNTU_VERSION=ubuntu1804
+    - name: CUDA 10.1
+      env:
+        - CUDA=10.1.168-1
+        - CUDA_SHORT=10.1
+        - UBUNTU_VERSION=ubuntu1804
+
+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}
+  - sudo apt update -qq
+  - sudo apt install -y cuda-${CUDA_SHORT/./-}
+  - 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
-  - pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
-  - pip install pycodestyle
-  - pip install flake8
-  - pip install codecov
-  - pip install sphinx
-  - pip install sphinx_rtd_theme
-  - pip install sphinx-autodoc-typehints
+  - pip install torch
+
 script:
+  - nvcc --version
   - python -c "import torch; print(torch.__version__)"
-  - pycodestyle .
-  - flake8 .
   - python setup.py install
-  - python setup.py test
-  - cd docs && make clean && make html && make doctest && cd ..
-after_success:
-  - codecov
-notifications:
-  email: false