Skip to content

Commit 2629151

Browse files
committed
[ci] cuda 10 changed to standard package names
1 parent 2bb8325 commit 2629151

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

.travis.yml

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,12 @@ install:
6262
- sudo dpkg -i $CUDA_REPO_PKG
6363
- rm ${CUDA_REPO_PKG}
6464
- travis_retry sudo apt-get -y update
65-
- 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
65+
- >
66+
if [ ${CUDA_VERSION_MAJOR} -lt 10 ]; then
67+
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
68+
else
69+
travis_retry sudo apt-get install -y --no-install-recommends --allow-unauthenticated libcuda-dev libcudart-dev libcublas-dev libcurand-dev
70+
fi
6671
- sudo ln -s /usr/local/cuda-${CUDA_VERSION_MAJOR}.${CUDA_VERSION_MINOR} /usr/local/cuda
6772

6873
before_script:

0 commit comments

Comments
 (0)