Update CUDA to version 6.5 in the Travis install script

BVLC · Sep 13, 2014 · 431a516 · 431a516
1 parent c69b3b4
commit 431a516
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/travis/travis_install.sh b/scripts/travis/travis_install.sh
@@ -57,19 +57,19 @@ popd
 
 # Install CUDA, if needed
 if $WITH_CUDA; then
-  CUDA_URL=http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1204/x86_64/cuda-repo-ubuntu1204_6.0-37_amd64.deb
+  CUDA_URL=http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1204/x86_64/cuda-repo-ubuntu1204_6.5-14_amd64.deb
   CUDA_FILE=/tmp/cuda_install.deb
   curl $CUDA_URL -o $CUDA_FILE
   dpkg -i $CUDA_FILE
   rm -f $CUDA_FILE
   apt-get -y update
   # Install the minimal CUDA subpackages required to test Caffe build.
   # For a full CUDA installation, add 'cuda' to the list of packages.
-  apt-get -y install cuda-core-6-0 cuda-extra-libs-6-0
+  apt-get -y install cuda-core-6-5 cuda-cublas-6-5 cuda-cublas-dev-6-5 cuda-cudart-6-5 cuda-cudart-dev-6-5 cuda-curand-6-5 cuda-curand-dev-6-5
   # Create CUDA symlink at /usr/local/cuda
   # (This would normally be created by the CUDA installer, but we create it
   # manually since we did a partial installation.)
-  ln -s /usr/local/cuda-6.0 /usr/local/cuda
+  ln -s /usr/local/cuda-6.5 /usr/local/cuda
 fi
 
 # Install LMDB