[ci] cublas changed the name!

Author: simogasp

.travis.yml

@@ -63,13 +63,14 @@ install:
   - sudo dpkg -i $CUDA_REPO_PKG
   - rm ${CUDA_REPO_PKG}
   - travis_retry sudo apt-get -y update
-  - travis_retry sudo apt-get install -y --no-install-recommends --allow-unauthenticated cuda-core-$CUDA_PKG_VERSION  cuda-cudart-dev-$CUDA_PKG_VERSION  cuda-cublas-dev-$CUDA_PKG_VERSION cuda-curand-dev-$CUDA_PKG_VERSION
-#  - >
-#    if [ ${CUDA_VERSION_MAJOR} -lt 10 ]; then
-#        travis_retry sudo apt-get install -y --no-install-recommends --allow-unauthenticated cuda-core-$CUDA_PKG_VERSION  cuda-cudart-dev-$CUDA_PKG_VERSION  cuda-cublas-dev-$CUDA_PKG_VERSION cuda-curand-dev-$CUDA_PKG_VERSION
-#    else
-#        travis_retry sudo apt-get install -y --no-install-recommends --allow-unauthenticated libcuda-dev  libcudart-dev libcublas-dev libcurand-dev
-#    fi
+  # cuda > 10.0 changed cublas naming
+  - >
+    if [ ${CUDA_VERSION_MAJOR} -lt 10 ]; then
+        CUBLAS_PKG=cuda-cublas-dev-$CUDA_PKG_VERSION
+    else
+        CUBLAS_PKG=libcublas-dev
+    fi
+  - travis_retry sudo apt-get install -y --no-install-recommends --allow-unauthenticated cuda-core-$CUDA_PKG_VERSION  cuda-cudart-dev-$CUDA_PKG_VERSION  ${CUBLAS_PKG} cuda-curand-dev-$CUDA_PKG_VERSION
   - sudo ln -s /usr/local/cuda-${CUDA_VERSION_MAJOR}.${CUDA_VERSION_MINOR} /usr/local/cuda
 
 before_script: