5,332
edits
Line 1: | Line 1: | ||
==Installation== | ==Installation== | ||
===Ubuntu | ===Ubuntu=== | ||
[https://developer.nvidia.com/cuda-toolkit CUDA Toolkit] | [https://developer.nvidia.com/cuda-toolkit CUDA Toolkit] | ||
See [https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#ubuntu-installation CUDA Ubuntu Installation] | See [https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#ubuntu-installation CUDA Ubuntu Installation] | ||
<syntaxhighlight lang="bash"> | <syntaxhighlight lang="bash"> | ||
# Set UBUNTU_VERSION to 2004 or 2204 | |||
UBUNTU_VERSION=$(lsb_release -sr | sed -e 's/\.//g') | |||
# Add NVIDIA package repositories | # Add NVIDIA package repositories | ||
wget https://developer.download.nvidia.com/compute/cuda/repos/ | wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/x86_64/cuda-ubuntu${UBUNTU_VERSION}.pin | ||
sudo mv cuda- | sudo mv cuda-ubuntu${UBUNTU_VERSION}.pin /etc/apt/preferences.d/cuda-repository-pin-600 | ||
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ | sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/x86_64/3bf863cc.pub | ||
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ | sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/x86_64/ /" | ||
# Install NVIDIA driver and cuda. | # Install NVIDIA driver and cuda. | ||
Line 22: | Line 20: | ||
sudo reboot | sudo reboot | ||
# Install cudnn | # Install cudnn | ||
sudo apt install libcudnn8 libcudnn8-dev | sudo apt install libcudnn8 libcudnn8-dev | ||
</syntaxhighlight> | </syntaxhighlight> |