5,323
edits
(→Conda) |
(→Ubuntu) |
||
Line 22: | Line 22: | ||
# Set UBUNTU_VERSION to 2004 or 2204 | # Set UBUNTU_VERSION to 2004 or 2204 | ||
UBUNTU_VERSION=$(lsb_release -sr | sed -e 's/\.//g') | UBUNTU_VERSION=$(lsb_release -sr | sed -e 's/\.//g') | ||
# Install nvidia driver | |||
sudo apt install nvidia-driver-530 | |||
# Add NVIDIA package repositories | # Add NVIDIA package repositories | ||
Line 29: | Line 32: | ||
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/x86_64/ /" | sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/x86_64/ /" | ||
# Install | # Install cuda. | ||
sudo apt install | sudo apt install cuda | ||
# Reboot and check that the drivers are working with nvidia-smi | # Reboot and check that the drivers are working with nvidia-smi | ||
sudo reboot | sudo reboot |