diff --git a/.travis.yml b/.travis.yml
index 5d53d790222741c1e1494762dd7215d45d6bdea4..c00b6cdc99bf44d9157088c0f1a3dfa5f9301bcb 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,4 +1,4 @@
-dist: xenial
+dist: bionic  # ubuntu 18.04
 language: python
 
 python:
@@ -6,7 +6,7 @@ python:
   - "3.6"
   - "3.7"
 
-env: CUDA=9.2.148-1 CUDA_SHORT=9.2 UBUNTU_VERSION=ubuntu1604
+env: CUDA=10.1.105-1 CUDA_SHORT=10.1 UBUNTU_VERSION=ubuntu1804
 
 # Ref to CUDA installation in Travis: https://github.com/jeremad/cuda-travis
 before_install:
@@ -23,7 +23,7 @@ before_install:
   - PATH=${CUDA_HOME}/bin:${PATH}
 
 install:
-  - pip install Cython
+  - pip install Cython torch==1.2
   - pip install -r requirements.txt
   - pip install -r tests/requirements.txt