一:驱动装置
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.debsudo dpkg -i cuda-keyring_1.0-1_all.debsudo 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.runsudo 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 在开端加上# cudaexport LD_LIBRARY_PATH=/usr/local/cuda/lib64export PATH=$PATH:/usr/local/cuda/bin而后按:wq保留退出编辑模式。9.刷新~/.bashrc文件使其失效:source ~/.bashrc10.从新应用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 update5.最初是真正开始装置cuDNN的库:sudo apt-get install libcudnn8=8.7.0.84-1+cuda11.8sudo apt-get install libcudnn8-dev=8.7.0.84-1+cuda11.8sudo apt-get install libcudnn8-samples=8.7.0.84-1+cuda11.8
6.查看cudnn装置胜利cd /usr/src/cudnn_samples_v8/mnistCUDNNsudo make cleansudo make呈现如下报错:
执行sudo apt-get install libfreeimage3 libfreeimage-devsudo make./mnistCUDNN呈现下图代表胜利