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