File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff 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
6873before_script :
You can’t perform that action at this time.
0 commit comments