CUDA: Difference between revisions
Line 23: | Line 23: | ||
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 | # Set UBUNTU_VERSION to 2004 or 2204 or 2404 | ||
UBUNTU_VERSION=$(lsb_release -sr | sed -e 's/\.//g') | UBUNTU_VERSION=$(lsb_release -sr | sed -e 's/\.//g') | ||
Line 29: | Line 29: | ||
sudo apt install nvidia-driver-565 | sudo apt install nvidia-driver-565 | ||
</syntaxhighlight> | </syntaxhighlight> | ||
;Notes | ;Notes | ||
* For machine learning, use Anaconda or Docker's CUDA since different versions of TensorFlow and PyTorch require different CUDA versions. | * For machine learning, use Anaconda or Docker's CUDA since different versions of TensorFlow and PyTorch require different CUDA versions. | ||
}} | }} |