在实际工作中，经常需要接触服务器，有时候在实行部署的时候需要对裸机举行安装和部署，涉及到深度学习环境搭建的话就会比较耗时一些，为了简化操纵工作以及保证与开发好的模型完美兼容最好的办法就是与我们自己的开发机器保持完全相同的环境版本，这时候就需要查询开发机器上的操纵环境，本文主要是在实际使用的时候记录一下CUDA和cuDNN内容的查询操纵。
1、查看 CUDA 版本
实行下面的命令来举行查询：
或者：
效果如下所示：
2、查看 cuDNN 版本
实行下面的命令：
- cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
- cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
- No such file or directory
- #define CUDNN_MAJOR 7
- #define CUDNN_MINOR 4
- #define CUDNN_PATCHLEVEL 2
- --
- #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
- #include "driver_types.h"
3、查看 GPU 信息
实行下面的命令可以完整查看GPU的详情信息：
效果如下所示：
这内里可以看到我们使用的驱动版本是：450.80.02
如果无法通过 cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 和 ldconfig -p | grep cudnn 命令精确显示 cuDNN 的版本信息，可以尝试以下几种方法来获取 cuDNN 的版本信息：
方法 1: 查抄 cuDNN 安装路径
首先，确认 cuDNN 的安装路径。通常，cuDNN 的库文件和头文件会安装在 /usr/local/cuda/include 和 /usr/local/cuda/lib64 目录下。
方法 2: 使用 dpkg 或 rpm 命令（实用于基于 Debian 或 Red Hat 的系统）
如果你是通过包管理器（如 apt 或 yum）安装的 cuDNN，可以使用以下命令来查看 cuDNN 的版本信息：
对于基于 Debian 的系统（如 Ubuntu）：
对于基于 Red Hat 的系统（如 CentOS）：
方法 3: 手动查找 cuDNN 版本信息
如果上述方法都无法获取 cuDNN 版本信息，可以手动查找 cuDNN 的版本信息文件。通常，cuDNN 的版本信息会包含在以下文件中：
- cuDNN 头文件：
- cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
- cuDNN 库文件：
- strings /usr/local/cuda/lib64/libcudnn.so | grep CUDNN_MAJOR
方法 4: 使用 nvcc 编译测试程序
你可以编写一个简单的 C++ 程序来查抄 cuDNN 版本信息：
- 创建一个 C++ 文件（例如 cudnn_version.cpp）：
- #include <iostream>
- #include <cudnn.h>
- int main() {
- std::cout << "cuDNN Version: " << CUDNN_MAJOR << "." << CUDNN_MINOR << "." << CUDNN_PATCHLEVEL << std::endl;
- return 0;
- }
- 编译并运行程序：
- nvcc -o cudnn_version cudnn_version.cpp -lcudnn
- ./cudnn_version
通过以上方法，应该可以或许获取 cuDNN 的版本信息。
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。 |
