diff --git a/.travis.yml b/.travis.yml
index d44c8519c4..07ad0e9a85 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -22,7 +22,7 @@ cache:
     directories:
         - $HOME/.local
 before_install:
-    - if [[ "$CUDA" = ON ]]; then source scripts/install-cuda-trusty.sh; fi
+    - if [ "$CUDA" = ON ]; then . scripts/install-cuda-ubuntu1604.sh; fi
     - if [ "$TRAVIS_OS_NAME" = linux ]; then scripts/install_cmake.sh; fi
     - sudo pip install requests
 script:
diff --git a/scripts/install-cuda-trusty.sh b/scripts/install-cuda-ubuntu1604.sh
similarity index 54%
rename from scripts/install-cuda-trusty.sh
rename to scripts/install-cuda-ubuntu1604.sh
index 6a6df53d41..69e7451335 100755
--- a/scripts/install-cuda-trusty.sh
+++ b/scripts/install-cuda-ubuntu1604.sh
@@ -8,13 +8,16 @@
 #
 # Taken from https://github.com/tmcdonell/travis-scripts.
 
-export CUDA_VER=8.0.61-1
-travis_retry wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_${CUDA_VER}_amd64.deb
-travis_retry sudo dpkg -i cuda-repo-ubuntu1404_${CUDA_VER}_amd64.deb
-travis_retry sudo apt-get update -qq
+set -e
+
+export CUDA_VER=9.0.176-1
+sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
+wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_${CUDA_VER}_amd64.deb
+sudo dpkg -i cuda-repo-ubuntu1604_${CUDA_VER}_amd64.deb
+sudo apt-get update -qq
 export CUDA_APT=${CUDA_VER:0:3}
 export CUDA_APT=${CUDA_APT/./-}
-travis_retry sudo apt-get install -qy cuda-core-${CUDA_APT} cuda-cudart-dev-${CUDA_APT}
-travis_retry sudo apt-get clean
+sudo apt-get install -qy cuda-core-${CUDA_APT} cuda-cudart-dev-${CUDA_APT}
+sudo apt-get clean
 export CUDA_HOME=/usr/local/cuda-${CUDA_VER:0:3}
 export PATH=${CUDA_HOME}/bin:${PATH}