nvidia-smi nvcc -V 及 CUDA、cuDNN 安装

泠山

已于 2023-09-27 20:00:34 修改

阅读量4.2k

点赞数 3

分类专栏：深度学习文章标签： CUDA python GPU 深度学习

于 2023-09-02 16:34:00 首次发布

本文链接：https://blog.csdn.net/qq_28087491/article/details/132635794

版权

8 篇文章 0 订阅

订阅专栏

相关文章：

1. 问题缘由

之前查找 CUDA 版本时都是直接使用的 nvidia-smi 指令，可以看到当前的 CUDA 版本是 12.0。
在这里插入图片描述

这时在使用 PyTorch 的时候，默认安装的与该 CUDA 相同版本的 PyTorch。

但是后面在编译程序的时候，问题就出现了程序内会报 CUDA_TOOLKIT_ROOT_DIR not found or specified 的错误，有些摸不着头脑。

然后就发现使用 nvcc-V 命令查看的时候，是找不到 CUDA 的。
在这里插入图片描述

这是因为 CUDA 有两种 API：运行时的 API 和驱动 API，即所谓的 Runtime API 与 Driver API。nvidia-smi 的结果除了有 GPU 驱动版本型号，还有 CUDA Driver API 的型号，即上面的 12.0。

而 nvcc-V 的结果是对应 CUDA Runtime API。而这时的 CUDA Runtime API 还没有安装。

如果想更新或安装 nvidia-smi 上的 CUDA 版本，最简单的方式是直接打开 Software & Updates，点击 Additional Drivers
在这里插入图片描述
选择一个驱动版本即可。

在官网下载能看见更直观的 CUDA 版本：https://www.nvidia.cn/geforce/drivers/
在这里插入图片描述

选择对应的显卡款式安装即可。

命令行键入 nvidia-smi 可以看到系统支持的最大CUDA版本，如果没有就不用管。因为上面截图过了这里就不放截图了。因为上面显示版本为 12.0，所以这里安装的 CUDA 最大版本应该是 12.0。
在英伟达官网找到想要安装的 CUDA 版本：https://developer.nvidia.com/cuda-11.1.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal

在这里插入图片描述

依照上面流程，输入：

sudo sh cuda_11.1.0_455.23.05_linux.run
键入accept：
注意，这个步骤如果前面安装了 Driver，即步骤1内显示了版本时，这里不要安装 Driver，否则这里的 CUDA 会安装失败；如果步骤1未安装，这里可以勾选。（这里我的.run软件不小心删了，拿的别人的，所以除了版本均一致）

安装成功后，打开.bashrc 或 .zshrc，写入以下信息：

export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

在下载 PyTorch 时，这里的 Compute Platform 是运行的平台，这里是不包括 CUDA 安装的。
在这里插入图片描述

cuDNN 地址如下 https://developer.nvidia.com/rdp/cudnn-download，不过要注意的是，需要注册一个账号，才能进入下载界面，建议下载 tar 的：

在这里插入图片描述

关注