5,323
edits
CUDA: Difference between revisions
→Ubuntu
(→Conda) |
(→Ubuntu) |
||
| Line 27: | Line 27: | ||
# Install nvidia driver | # Install nvidia driver | ||
sudo apt install nvidia-driver- | sudo apt install nvidia-driver-545 | ||
# Add NVIDIA package repositories | # Add NVIDIA package repositories | ||
| Line 40: | Line 40: | ||
sudo reboot | sudo reboot | ||
# Install cudnn | # Install cudnn if needed | ||
sudo apt install libcudnn8 libcudnn8-dev | sudo apt install libcudnn8 libcudnn8-dev | ||
</syntaxhighlight> | </syntaxhighlight> | ||