0


笔记--Ubuntu20.04安装Nvidia驱动、CUDA Toolkit和CUDA CuDNN

1--安装Nvidia驱动

① 查看可安装的Nvidia驱动版本:

ubuntu-drivers devices

② 安装相应版本的Nvidia驱动:

博主这里选择的是第一个,也可以安装推荐(recommended)的版本

sudo apt-get install nvidia-driver-515

安装过程中,一般要设置一个密码,这个密码在后面重启系统会用到!

③ 安装完成后,需要重启电脑:

reboot

重启电脑一般会弹出以下Perform MDK management界面:

选择第二个选项 Eroll MDK,输入第②步设置的密码后,会发现只剩下三个选项,选择第一个开机选项即可完成 Nvidia 驱动的安装;

④ 验证安装:

nvidia-smi

出现以上界面即为安装成功!

2--安装CUDA

2-1--禁用nouveau

① 查看黑名单

sudo gedit /etc/modprobe.d/blacklist.conf

② 在弹出的文本末尾添加黑名单

blacklist nouveau

③ 保存并重启

sudo update-initramfs -u
 
reboot

④ 测试是否禁用成功

lsmod | grep nouveau

无输出即禁用 nouveau 成功!

2-2--选择CUDA Toolkit

NVIDIA CUDA驱动版本下载地址:https://developer.nvidia.com/cuda-toolkit-archive

注:选择合适的版本进行安装,确保 CUDA Toolkit 的版本低于 Nvidia 驱动的版本!(这里博主的CUDA Toolkit 版本为465.19.01,低于安装的 Nvidia驱动 版本515.86.01,可以兼容和适配)

2-3--下载和安装CUDA Toolkit

① 根据2-2截图中的两条命令进行安装(第一个命令将下载CUDA Toolkit,第二个命令将安装CUDA Toolkit):

wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run

sudo sh cuda_11.3.1_465.19.01_linux.run

在执行安装命令的时候,也许需要先执行以下命令:

sudo chmod +x cuda_11.3.1_465.19.01_linux.run

再执行安装命令:

sudo sh cuda_11.3.1_465.19.01_linux.run

accept 之后,不要安装驱动,因为之前我们已经安装了Nvidia驱动!

2-4--配置环境变量

sudo gedit ~/.bashrc

末尾添加以下两条路径:(由于博主安装的是 Cuda11.3,所以路径为cuda-11.3,这里需根据实际的安装版本和路径进行设定)

export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新系统变量:

source ~/.bashrc

2-5--测试是否安装成功:

nvcc -V

3--安装CUDA CuDNN

CUDA CuDNN下载地址:https://developer.nvidia.com/rdp/cudnn-archive

① 下载合适的CUDA CuDNN版本:(选择第二项x86 64进行下载)

② 解压安装包:

tar -xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz

③ 拷贝文件到cuda安装目录,并给予执行权限:

sudo cp -d cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp -d cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

④ 测试是否安装成功:

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

输出对应的版本号即为安装成功!

4--测试pytorch能否使用Cuda

pytorch下载地址:https://pytorch.org/

① 下载合适版本的 pytorch(这里博主选用的是基于 Cuda11.3 的 v1.12.0 版本);

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch

② 验证Cuda可用性:

python

import torch

print(torch.cuda.is_available())

当输出为** True** 时,表明 pytorch 安装成功,pytorch 可以使用 Cuda 进行加速,Nvidia驱动、CUDA Toolkit 和 CUDA CuDNN 等均安装成功!!!!


本文转载自: https://blog.csdn.net/weixin_43863869/article/details/128267988
版权归原作者 憨豆的小泰迪 所有, 如有侵权,请联系我们删除。

“笔记--Ubuntu20.04安装Nvidia驱动、CUDA Toolkit和CUDA CuDNN”的评论:

还没有评论