Ubuntu22.04中安装cuda, cudnn, pytorch
Ubuntu 22.04中安装Nvidia显卡驱动，cuda, cudnn, pytorch扼要介绍怎样一步步安装，直到能用pytorch跑深度学习代码。紧张分成四大步：1. 安装Nvidia显卡驱动 2. 安装cuda 3. 安装cudnn 4. 安装pytorch
一、 安装Nvidia显卡驱动
步调 1: 登录Nvidia网站（https://www.nvidia.com/download/index.aspx），搜索本身须要的显卡驱动进行安装即可（这一步请大家自行查找得当本身显卡的驱动，自行安装。）
https://i-blog.csdnimg.cn/direct/627efad57051449cadc4b1cd6555aa68.png
Fig. 1 Nvidia显卡驱动网站
步调 2：在终端输入下令 nvidia-smi 检查显卡驱动是否存在，请见下图Fig. 2所示，
https://i-blog.csdnimg.cn/direct/27f73569c08c41bca265fc0cb0ea9ed4.png
Fig. 2 显卡驱动信息
在Fig. 2中，我们可以看到系统中存在的Nvidia显卡驱动为535.183.01（恭喜，Nvidia显卡驱动安装成功；如果发现没有驱动，须要重新操纵来保证Nvidia显卡驱动的成功安装），旁边显示的CUDA Version: 12.2是这个Nvidia显卡驱动版本535.183.01兼容的cuda的最高版本。也就是说，我们在当前显卡驱动版本下，可以安装cuda 12.2, 也可以安装更低的cuda版本，如cuda 12.1或者cuda 11.8。
二、 安装cuda
有了上述Fig. 2提供的cuda版本信息，我们可以进入Nvidia网站搜索合适的cuda版本，来进行安装。首先辈入下图Fig. 3所示的Nvidia CUDA Toolkit网站（https://developer.nvidia.com/cuda-downloads），
https://i-blog.csdnimg.cn/direct/6a8c8b38da3444c6b089c3f51589b8b3.png
Fig. 3 Nvidia的cuda安装网站
在Fig. 3中，点击“Archive of Previous CUDA Releases”进入cuda汗青版本的页面，探求合适的cuda版本。点击后，进入下图Fig. 4所示的网页，
https://i-blog.csdnimg.cn/direct/45ca87eb50324870a042fcf6d551292c.png
Fig. 4 选择cuda 11.8版本
在Fig. 4中，选择cuda 11.8版本（与我们上述的Nvidia显卡驱动版本兼容，因为我们的Nvidia显卡驱动支持最高到12.2的cuda版本）。点击“CUDA Toolkit 11.8.0”， 进入如下图Fig. 5所示的网页，
https://i-blog.csdnimg.cn/direct/e4c3d0a10c074628b7819d4c74e2e295.png
Fig. 5 选择系统种别和版本等信息
在Fig. 5中，我们按照本身电脑的实际环境，按照顺序选择，
（1）Linux
（2）x86_64（我们可以在终端输入下令uname -m来查看电脑的体系架构，我的显示结果的是x86_64）
（3）Ubuntu
（4） 22.04 (根据本身的Ubuntu版本进行选择)
（5）runfile (选择这个格式的文件安装起来很方便，只须要使用Fig. 5图片下方主动天生的两行下令进行下载和安装即可)
在Fig. 5中，在我们选择了操纵系统类型（Linux），系统体系架构（x86_64），系统种别（Ubuntu），系统版本（22.04）和安装文件类型（runfile）后，主动天生了如Fig. 5图片下方天生的两行安装下令，我们在终端先运行第一行下令（下令如下所示），运行结果如Fig. 6所示。
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
https://i-blog.csdnimg.cn/direct/772e5bd5cef94ef3b4f334b9d952f8ff.png
Fig. 6 第一行下令运行结果截图
然后我们运行第二行下令（下令如下所示），
sudo sh cuda_11.8.0_520.61.05_linux.run
在运行这行下令的时间，我们须要手动选择一些安装选项，如图Fig. 7(a)，Fig. 7(b)和Fig. 7©所示，
https://i-blog.csdnimg.cn/direct/9aa2617e0aef46718932064aa7c8b9c6.png
Fig. 7(a) 安装选项一
如图Fig. 7(a)所示，请选择“Continue”选项。回车进入下一个界面。
https://i-blog.csdnimg.cn/direct/d2210f5417ab4ff490d6cc2ed4edfbea.png
Fig. 7(b) 安装选项二
如图Fig. 7(b)所示，请输入“accept”。回车进入下一个界面。
https://i-blog.csdnimg.cn/direct/0373f13a56584fc4bf59b634679e31ee.png
Fig. 7(c ) 安装选项三
如图Fig. 7(c )所示，不要选择上面的Driver（勾选这个意味着我们在安装这个cuda的时间，同时也会给系统安装兼容当前要安装的cuda版本的Nvidia显卡驱动。我们之前已经安装过了Nvidia显卡驱动，这里再安装就辩论了），这里只勾选图中“黄色”字样的信息即可。末了，请选择“Install”进行安装。安装完成后，我们的终端如图Fig. 7(d)所示。
https://i-blog.csdnimg.cn/direct/1557a52debce4d8886180c84217585c6.png
Fig. 7(d) 第二行下令运行结果
按照上述操纵，安装完成cuda后，我们可以在系统中/usr/local/路径下，看到cuda-11.8和cuda文件夹，如图Fig. 8所示。
https://i-blog.csdnimg.cn/direct/40d99a607a4743b8a5cfc9bfe0439a31.png
Fig. 8 cuda安装成功后的系统文件的变化
然后，我们须要在Ubuntu系统中添加关于cuda的环境变量（ps: 添加环境变量的意义，每次打开终端，我们可以在终端中使用环境变量中相干的下令；否则，在终端中输入有关cuda的下令，终端找不到下令相干的文件，就会报错执行不了；比如在这里，我们添加在系统中添加关于cuda的环境变量后，就可以在终端输入关于cuda的下令nvcc -V
来查看系统中cuda的版本。简而言之，设置环境变量的作用就是帮助系统找到终端中的输入下令的对应文件的路径）。
具体添加环境变量的流程：
[*]在终端中输入如下下令（用于打开系统存储环境变量的.bashrc文件）
gedit ~/.bashrc
[*]在.bashrc文件中添加如下指令（把如下下令复制粘贴到.bashrc文件中即可，如下图Fig. 9所示，添加后请记得保存）
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
https://i-blog.csdnimg.cn/direct/a5683362085840ac8d26fbca96c98686.png
Fig. 9 .bashrc文件中添加cuda环境变量
[*]在终端中输入如下下令使得.bashrc文件中的cuda环境变量起作用
source ~/.bashrc
[*]在终端中输入如下下令查看系统中的cuda版本（结果如图Fig. 10所示）。
nvcc -V
https://i-blog.csdnimg.cn/direct/1416ff2926f14e61ad5a8f83ab72322f.png
Fig. 10 查看系统中的cuda版本
到这里，我们的cuda也安装成功了。
三、安装cudnn
cuDNN（CUDA Deep Neural Network library）是由 NVIDIA 提供的一个 GPU 加速库，专门用于深度神经网络的高性能盘算。它提供了一组高度优化的例程，用于深度学习的各种操纵，如卷积、池化、归一化和激活函数。cuDNN 紧张用于深度学习框架（如 TensorFlow、PyTorch、Caffe 和 MXNet）中，以提高盘算效率和性能。（这段话是ChatGPT说的）
言归正传，成功安装cuda后，我们可以进入Nvidia网站搜索合适的cuDNN版本，来进行安装。首先辈入下图Fig. 11所示的Nvidia cuDNN网站（https://developer.nvidia.com/cudnn-downloads），
https://i-blog.csdnimg.cn/direct/31b9add249ef47d8804f2825b9ec7c91.png
Fig. 11 Nvidia的cuDNN安装网站
在Fig. 11中，点击“Archive of Previous CUDA Releases”进入cuDNN汗青版本的页面，探求合适的cuDNN版本。点击后，进入下图Fig. 12 (a)所示的网页，再进入Fig. 12 (b)所示的网页，
https://i-blog.csdnimg.cn/direct/5b5e63e0aa1d409c9a7254232469d8ca.png
Fig. 12(a) cuDNN网站界面一
https://i-blog.csdnimg.cn/direct/625a488d3c7343fcbae744536ccdf1ba.png
Fig. 12(b) cuDNN网站界面二
https://i-blog.csdnimg.cn/direct/d299ee924a88441c97f4770b310ccdea.png
Fig. 12 (c ) cuDNN网站界面三
https://i-blog.csdnimg.cn/direct/d2a67e810508400683c01b27ffc9c0ee.png
Fig. 13 Nvidia网站中的cuDNN的tar格式文件的安装下令
在图Fig. 13中，Nvidia官方提供的安装分为三个步调，分别如下。
[*]将在终端将路径切换到下载的.tar.xz安装包的路径下， 下载的安装包在如图Fig. 14所示的文件夹中，
cd /home/nearlab-iiwa/Downloads
https://i-blog.csdnimg.cn/direct/dcf7e6e544d24b1fb7fe633cee8974f8.png
Fig. 14 cuDNN压缩包
[*]解压文件下令（注意：如Fig. 13中所述，须要把这个下令中的解压的文件名修改一下，最简朴的方式是，用下载的压缩包的文件名替换tar -xvf反面的文件名就行了）
tar -xvf cudnn-linux-x86_64-8.9.5.30_cuda11-archive.tar.xz
[*]将cuDNN解压出来的文件拷贝到之前安装的cuda文件加中（cuda文件夹的路径如Fig. 8和Fig. 9中所示），然后给cudnn和cudnnlib等文件读取权限。具体下令为下面三行。
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
按照这三个步调操纵完成后，可以发现，cuda文件夹（路径一样平常为/usr/local/）下的cuda/include文件夹中，多了一些以cudnn开头的头文件，如Fig. 15所示。同样地，cuda/lib64/文件夹中，多了一些以libcudnn开头的文件。
https://i-blog.csdnimg.cn/direct/67ebce01d32243f59935c4e4a74ee603.png
Fig. 15 /usr/local/cuda/include文件夹中的变化
https://i-blog.csdnimg.cn/direct/24348644842b4d829b4091ea6f9ec80f.png
Fig. 16 /usr/local/cuda/lib64文件夹中的变化
安装完成后，可以在终端输入如下下令，验证cuDNN是否在系统中成功安装。运行后的结果如图Fig. 17所示。
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
https://i-blog.csdnimg.cn/direct/45f5741c59ca45ef939c7aa77d00ca30.png
Fig. 17 验证系统中的cuDNN
至此，cuDNN安装成功。
四、安装pytorch
进入Pytorch官网（https://pytorch.org/）。
https://i-blog.csdnimg.cn/direct/2cb75a517a144dad88642fc558dd4d99.png
Fig. 18 pytorch界面
如下图Fig. 18所示选择构建方式（Stable），操纵系统（Linux），包管理工具（pip），编程语言（Python），盘算硬件（cuda 11.8）后，界面主动天生了安装下令。安装下令如下所示（这里使用了包管理工具pip3，如果系统中没有这个下令，可以先在终端使用下令 sudo apt install python3-pip 来安装），
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
安装完成后，可以使用如下下令来检查pytorch是否安装成功，cuda是否可用。
python3
import torch
print(torch.__version__)
print(torch.cuda.is_available())
分别在终端输入上述下令后，输出结果如下图Fig. 19所示，torch.cuda.is_available()的结果为True说明cuda可用。
https://i-blog.csdnimg.cn/direct/ab4720b5ccbc48529f1b0bc3f03f2813.png
五、致谢
我在安装cuda，cuDNN，pytorch的时间，参考了一些写的很不错的博客，在此表示诚挚的感谢。下面把这些博客列在下面，以供大家参考。
Ubuntu22.04安装CUDA和cuDNN详细过程记录
Ubuntu20.04下CUDA、cuDNN的详细安装与设置过程（图文）
【cuda\cudnn安装教程】以及环境变量设置(以cuda11.8为例)
新装UBUNTU 22.04安装Nvidia GPU版Pytorch完整教程
写在末了
局限于本人能力，如果本博客中有任何题目，请大家批评指正，不胜感激。如果有安装过程的题目，也可以留言交流。
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
页:
[1]