diff --git a/tests/install_travis.sh b/tests/install_travis.sh
index 8437a29a6ed04b68668c95b70bbecdad0c34f0f0..fda8ac3591d0b2aef0e195cdcab1f1dc55b0ba20 100755
--- a/tests/install_travis.sh
+++ b/tests/install_travis.sh
@@ -48,7 +48,7 @@ sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
 sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
 sudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
 sudo apt-get update
-sudo apt-get -y install cuda-$CUDA_VERSION
+sudo apt-get -y install cuda
 sudo ldconfig
 export CUDA_HOME=/usr/local/cuda
 export PATH=/usr/local/cuda/bin:$PATH