一:驱动安装
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
1.输入网址:
https://developer.nvidia.com/cuda-toolkit-archive
本文选择11.7.0版本的CUDA
2.安装cuda,选择相应版本和系统,并执行下面两行命令。如下图所示。
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
4.输入accept
5.Driver前输入空格。
6.点击install
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版本。
下图是cudnn安装的详细步骤。
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
出现如下报错:
执行
sudo apt-get install libfreeimage3 libfreeimage-dev
sudo make
./mnistCUDNN
出现下图代表成功
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。