Skip to content

Instantly share code, notes, and snippets.

@vaibhaw

vaibhaw/caffe-cuda-cudnn_install.sh

Last active July 7, 2016 12:29
Show Gist options
  • Save vaibhaw/7bbce14e15d55d50ae7d15bb4c0c1ea0 to your computer and use it in GitHub Desktop.
Save vaibhaw/7bbce14e15d55d50ae7d15bb4c0c1ea0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
caffe-cuda-cudnn_install.sh
sudo apt-get update -y
sudo apt-get install -y git wget linux-image-generic build-essential unzip
cd /tmp
# cuda downloads - http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_7.0-28_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1404_7.0-28_amd64.deb
sudo apt-get update && sudo apt-get install cuda-toolkit-7-0
echo -e "\nexport CUDA_HOME=/usr/local/cuda" >> ~/.bashrc
echo -e "\nexport CUDA_ROOT=/usr/local/cuda" >> ~/.bashrc
echo -e "\nexport PATH=/usr/local/cuda/bin:\$PATH" >> ~/.bashrc
echo -e "\nexport LD_LIBRARY_PATH=/usr/local/cuda/lib64:\$LD_LIBRARY_PATH" >> ~/.bashrc
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment