一:驱动装置
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
呈现下图代表胜利