Ubuntu 安装CUDA

CUDA安装

NVIDIA® GPU 驱动程序

https://www.nvidia.com/drivers
根据显卡型号下载 NVIDIA-Linux-x86_64-460.73.01.run

sudo chmod +x NVIDIA-Linux-x86_64-460.73.01.run
sudo ./NVIDIA-Linux-x86_64-460.73.01.run 

Ubuntu没必要安装这个版本，直接在“软件和更新”中选择NVIDIA的驱动更好

CUDA® 工具包

https://developer.nvidia.com/cuda-toolkit-archive
下载Ubuntu对于的包安装，sample没必要安装

sudo dpkg -i libcudnn8_8.2.0.53-1+cuda11.3_amd64.deb 
sudo dpkg -i libcudnn8-dev_8.2.0.53-1+cuda11.3_amd64.deb

设置环境变量

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64

测试

import tensorflow as tf
print(tf.test.is_gpu_available())

根据打印排查问题，如果最后输出为True 则安装成功；否则根据日志排查问题。

遇到的问题

问题1

ERROR: An NVIDIA kernel module ‘nvidia-drm’ appears to already be loaded in your kernel.
Linux 驱动已经安装，如果需要重新安装，在“软件和更新”中选择非NVIDIA的驱动，重启，再安装。其实没必要，直接使用“软件和更新”中的驱动更好

问题2

Could not load dynamic library ‘libcusolver.so.10’

cd $LD_LIBRARY_PATH
sudo ln libcusolver.so.11 libcusolver.so.10