ubuntu安装nvidia显卡的驱动和cuda
以前源码安装驱动和cuda的过程是比较繁琐的,现在可以直接使用apt去安装驱动已经cuda,具体的命令如下。
1. 安装英伟达显卡驱动
搜索显卡驱动
使用下面代码去搜索最新版本的驱动。
sudo apt-cache search nvidia-*
可以看到目前最新的版本是510,我们就可以安装510的驱动了,使用下面命令安装即可。
sudo apt install nvidia-driver-510
2. 安装cuda
CUDA是nvidia显卡编程的接口,现在安装cuda也同样简单,我们去官网就可以找到CUDA的安装代码。
https://developer.nvidia.com/cuda-downloads
依次选择: Linux – x86_64 – Ubuntu - 20.04 – deb(network)就可以看到如下安装代码了。查看安装的代码,我们看到,实际上是在apt上面导入了证书。
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt-get update
sudo apt-get -y install cuda
如果想指定cuda版本进行安装,把最后一行命令带上版本号,如安装11.3版本CUDA的命令:
sudo apt-get -y install cuda-11-3
添加CUDA的环境变量,需要把cuda添加到环境变量才能运行。 打开 ~/.bashrc文件,在文件末尾添加如下内容:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
生效环境变量:
source ~/.bashrc
测试cuda是否安装成功
nvcc -V
3. 安装cuDNN
cuDNN是一个GPU矩阵运算的加速包,其安装也很简单,只需要一个命令就可以完成了,如下:
sudo apt install libcudnn8 libcudnn8-dev
|