-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcuda-cudnn.sh
More file actions
17 lines (13 loc) · 750 Bytes
/
cuda-cudnn.sh
File metadata and controls
17 lines (13 loc) · 750 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
INSTALLDIR="/usr/local/cuda"
echo "${PACKAGE} ${VERSION} ${INSTALLDIR}"
echo "Installing CUDA and cuDNN"
echo "Installing CUDA"
cd ${INSTALLDIR}
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /"
sudo apt-get update -y
sudo apt-get install libcudnn8=8.5.0.*-1+cuda11.7
# sudo apt-get install libcudnn8-dev=${cudnn_version}-1+${cuda_version}
sudo apt-get install libcudnn8-dev=8.5.0.*-1+cuda11.7