Merge pull request BVLC#4793 from cypof/cudnn_path

NV changed path to cuDNN
howard0su · Oct 4, 2016 · a2179bd · a2179bd
2 parents a7f950b + 08ca703
commit a2179bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/travis/install-deps.sh b/scripts/travis/install-deps.sh
@@ -84,7 +84,7 @@ if $WITH_CUDA ; then
   rm $CUDA_REPO_PKG
 
   if $WITH_CUDNN ; then
-    ML_REPO_PKG=nvidia-machine-learning-repo_4.0-2_amd64.deb
+    ML_REPO_PKG=nvidia-machine-learning-repo-ubuntu1404_4.0-2_amd64.deb
     wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64/$ML_REPO_PKG
     dpkg -i $ML_REPO_PKG
   fi