Ubuntu 安装 cuDNN（附测试）

### 安装 CuDNN on Ubuntu 的逐步指南 为了在Ubuntu上成功安装CuDNN，需遵循一系列特定步骤来确保兼容性和性能优化。首先确认已安装NVIDIA驱动程序以及CUDA Toolkit版本匹配所要安装的CuDNN版本[^3]。 #### 验证 CUDA 版本 通过命令`nvcc --version`可以查看当前系统中的CUDA编译器版本，以此决定适合的CuDNN版本。这一步骤对于后续操作至关重要，因为不同版本间的兼容性可能影响到最终应用的表现效果[^4]。 ```bash $ nvcc --version ``` #### 下载 CuDNN 库文件 访问[NVIDIA官方网站](https://developer.nvidia.com/cudnn)，注册并登录账户后下载对应于已安装CUDA版本的CuDNN库压缩包。注意选择适用于Linux系统的tar格式文件进行下载[^5]。 #### 解压与复制文件至指定路径 解压下载好的cuDNN tarball文件，并将其内的头文件(`*.h`)、库文件(`lib*`)分别拷贝到对应的CUDA目录下： ```bash $ tar -xzvf cudnn-<version>-linux-x64-v8.0.tgz $ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include/ $ sudo cp cuda/lib64/* /usr/local/cuda/lib64/ ``` 完成上述动作之后记得执行环境变量更新指令使更改生效: ```bash $ ldconfig ``` #### 测试安装是否成功 最后可以通过编写简单的测试代码验证CuDNN是否被正确加载和初始化。创建一个新的C++源码文件test_cudnn.cpp，在其中调用cudnnCreate()函数尝试建立句柄对象；如果返回状态为CUDNN_STATUS_SUCCESS，则说明配置无误[^6]。 ```cpp #include <iostream> #include <cudnn.h> int main(){ cudnnHandle_t handle; cudnnStatus_t status = cudnnCreate(&handle); if (status != CUDNN_STATUS_SUCCESS){ std::cerr << "Failed to create cuDNN context." << std::endl; return 1; } std::cout << "Successfully created cuDNN context!" << std::endl; // Clean up resources before exit. cudnnDestroy(handle); return 0; } ``` 编译运行这段小程序即可得知结果。若有任何错误提示，请仔细检查之前的每一步设置过程直至解决问题为止。