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