显卡驱动,cuda,cudnn安装

一:驱动安装
1.安装驱动
选择阿里云镜像。
显卡驱动,cuda,cudnn安装_第1张图片
2.在终端输入以下命令:

sudo apt-get install linux-headers-$(uname -r)
distribution=$(. /etc/os-release;echo $ID$VERSION_ID | sed -e 's/\.//g')
wget https://developer.download.nvidia.com/compute/cuda/repos/$distribution/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring_1.0-1_all.deb
sudo apt-get -y install cuda-drivers

3.输入nvidia-smi 查看结果
显卡驱动,cuda,cudnn安装_第2张图片
二:安装CUDA

1.输入网址:
https://developer.nvidia.com/cuda-toolkit-archive
本文选择11.7.0版本的CUDA

显卡驱动,cuda,cudnn安装_第3张图片
2.安装cuda,选择相应版本和系统,并执行下面两行命令。如下图所示。
显卡驱动,cuda,cudnn安装_第4张图片

wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run

sudo sh cuda_11.7.0_515.43.04_linux.run

3.选择continue
显卡驱动,cuda,cudnn安装_第5张图片
4.输入accept
显卡驱动,cuda,cudnn安装_第6张图片
5.Driver前输入空格。
6.点击install
显卡驱动,cuda,cudnn安装_第7张图片

7.查看cuda  
cd /usr/local  看看有没有cuda文件夹
8.打开、编辑环境变量的配置文件
vi ~/.bashrc 
在末尾加上
# cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
然后按:wq保存退出编辑模式。
9.刷新~/.bashrc文件使其生效:
source ~/.bashrc
10.重新使用nvcc命令查看CUDA版本
nvcc -V
如下图cuda版本11.7

三:安装cudnn

1.先下载cuDNN的.deb本地安装包
进入网址:https://developer.nvidia.com/rdp/cudnn-archive
注意下载是需要注册一个Nvidia的账号的,本文选择下载cuDNNv8.7.0,因为其支持CUDA11版本。

显卡驱动,cuda,cudnn安装_第8张图片
下图是cudnn安装的详细步骤。
显卡驱动,cuda,cudnn安装_第9张图片

2.sudo dpkg -i cudnn-local-repo-ubuntu2204-8.7.0.84_1.0-1_amd64.deb 
3.sudo cp /var/cudnn-local-repo-ubuntu2204-8.7.0.84/cudnn-local-BF23AD8A-keyring.gpg /usr/share/keyrings/
4.更新一下软件库
sudo apt-get update
5.最后是真正开始安装cuDNN的库:
sudo apt-get install libcudnn8=8.7.0.84-1+cuda11.8
sudo apt-get install libcudnn8-dev=8.7.0.84-1+cuda11.8
sudo apt-get install libcudnn8-samples=8.7.0.84-1+cuda11.8
6.查看cudnn安装成功
cd /usr/src/cudnn_samples_v8/mnistCUDNN
sudo make clean
sudo make
出现如下报错:

显卡驱动,cuda,cudnn安装_第10张图片

执行
sudo apt-get install libfreeimage3 libfreeimage-dev
sudo make
./mnistCUDNN
出现下图代表成功

显卡驱动,cuda,cudnn安装_第11张图片

你可能感兴趣的:(ubuntu深度学习人脸识别)