CUDA、cuDNN安装

CUDA、cuDNN安装

(小白一枚,内容为记录安装配置步骤,仍在完善)

一、下载安装CUDA

1.在CUDA Toolkit Archive中选择对应的版本下载
比如我需要安装cuda10.1并且我的电脑系统为win10
那么进入点击进入对应版本后按以下勾选然后点击下载
在这里插入图片描述
2.选择自定义安装,安装组件全部勾选
在这里插入图片描述
3.安装位置可默认也可手动指定路径,但要记得自己安装的位置,因为后面需要配置系统环境变量
在这里插入图片描述
CUDA安装完成后,然后看看自己的环境变量,CUDA9.1版本安装之后会自动配置好环境变量,不需手动添加。
在这里插入图片描述
在这里插入图片描述
在cmd中输入nvcc -V查看安装的CUDA的版本
在这里插入图片描述

二、下载安装cuDNN

1.进入cuDNN Archive选择对应的版本下载
在这里插入图片描述
下载成功后解压cuDNN压缩包
在这里插入图片描述
将此目录下的所有文件复制到CUDA安装目录中
在这里插入图片描述

CUDAcuDNN 是深度学习模型训练过程中不可或缺的两个重要工具。CUDA(Compute Unified Device Architecture)是由 NVIDIA 开发的一种并行计算平台和 API 模型,它允许开发者利用 GPU 的强大性能加速应用程序;而 cuDNNCUDA Deep Neural Network library)则是专门为深度神经网络设计的一个高性能库,能够显著提升卷积、池化等操作的速度。 下面是关于如何安装 CUDAcuDNN 的详细步骤,并附带一些常见问题及解决办法。 ### 一、CUDA 安装 #### 1. 下载 CUDA Toolkit 前往 [NVIDIA 官方网站](https://developer.nvidia.com/cuda-downloads) 并选择合适的操作系统版本以及其他配置选项(比如架构)。根据你的硬件情况和个人需求挑选相应的 CUDA 版本下载链接。 #### 2. 配置系统环境变量 将 CUDA 工具包添加到系统的 PATH 环境变量中,以便可以在终端或命令提示符窗口直接访问 nvcc 编译器和其他工具。通常路径类似于 `/usr/local/cuda/bin` 或 `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vX.X\bin`。 此外,还需要设置 LD_LIBRARY_PATH(Linux/macOS)或者 %PATH%(Windows),使其包含 CUDA 库的位置,默认为 `/usr/local/cuda/lib64` 或者 `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vX.X\lib\x64`。 #### 3. 测试安装 打开一个新的命令行界面运行以下命令检查是否成功安装CUDA: ```sh nvcc --version ``` 你应该会看到类似如下的输出信息: ``` nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2020 NVIDIA Corporation Built on Mon_Oct_12_20:18:47_PDT_2020 Cuda compilation tools, release 11.1, V11.1.105 Build cuda_11.1.TC455_02.105-relaxed_cuda_deps ``` 这表示 NVCC 编译器可用,并且指出了当前使用的 CUDA 版本。 --- ### 二、cuDNN 安装 #### 1. 注册并登录 NVIDIA 账户 如果你还没有 NVIDIA Developer Program 成员资格,请先注册成为会员然后登录。 #### 2. 获取 cuDNN 文件 进入 [cuDNN 下载页面](https://developer.nvidia.com/rdp/cudnn-download),找到适用于已安装 CUDA 版本的那个 cuDNN 发行版,点击“Download”按钮获取压缩包文件。 #### 3. 解压与复制文件 解压所获得的 tar.gz (.tgz) 文件夹内容至指定目录。假设你选择了 Linux x86_64,则可以使用下列指令进行提取和移动必要的头文件(.h)和共享库(例如 .so): ```bash tar -xzvf cudnn-X.Y-linux-x64-vZ.tgz # 替换 X Y Z 表示具体的版本编号 sudo cp cuda/include/* /usr/local/cuda/include/ sudo cp cuda/lib64/* /usr/local/cuda/lib64/ ``` 对于 Windows 用户来说,只需将 bin, include, lib 文件夹的内容复制到相应位置即可(`C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vX.X`)。 #### 4. 更新动态链接器缓存(仅限于 Linux) 为了让程序能自动定位新加入的 cuDNN 共享对象,需刷新一次 ldconfig 数据库索引: ```bash sudo ldconfig ``` #### 5. 最后一步验证 现在我们来确保所有东西都按预期工作着——编写一段简短的例子代码来进行测试: ```cpp #include <iostream> #define CUDART_API __declspec(dllimport) extern "C" { // 导入 cuDNN 函数声明... } int main(){ std::cout << "If you see this message your installation is correct." << std::endl; return 0; } ``` 编译上面的小例子(记得加上 `-lcudnn` 参数以连接静态库),假如没有任何错误发生就说明整个过程顺利完成! --- **注意事项:** - 安装前建议备份现有数据以防万一; - 对于多用户服务器环境可能需要管理员权限才能修改全局环境变量; - 如果遇到兼容性问题,尽量保持各软件组件之间的版本一致性; - 即便是最新发布的 CUDA/cuDNN 组件也可能存在某些未知 bug,查阅官方论坛可以获得帮助和支持。 --
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值