Skip to content

Instantly share code, notes, and snippets.

@MittalShruti

MittalShruti/cudnn.sh

Created May 9, 2020 13:31
Show Gist options
  • Save MittalShruti/04abeb712f86226e62754073e691ef48 to your computer and use it in GitHub Desktop.
Save MittalShruti/04abeb712f86226e62754073e691ef48 to your computer and use it in GitHub Desktop.
Download ZIP
Get cuDNN
Raw
cudnn.sh
#Fetch cuDNN
CUDNN_TAR_FILE="cudnn-10.0-linux-x64-v7.6.5.32.tgz"
wget https://developer.nvidia.com/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.0_20191031/cudnn-10.0-linux-x64-v7.6.5.32.tgz
sudo tar -xzvf ${CUDNN_TAR_FILE}
#copy the files in cuda toolkit directory
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment