4个版本

0.1.3 2024年3月12日
0.1.2 2024年2月12日
0.1.1 2024年2月12日
0.1.0 2024年1月28日

机器学习 中排名第811

MIT 许可证

17KB
189

nvidia-checker-rs

nvidia-checker包是一个用于检查NVIDIA软件环境的命令行实用程序。当开发者想要使用NVIDIA软件的相同环境时可以使用它。

  • 支持的功能
    • 作为库使用
    • 检查指定版本
    • 检查指定范围内的版本
  • 文档

作为库使用

nvidia-checker = "0.1.2"

开始使用CLI

安装

  • 安装 cargo
  • 安装 nvidia-checker
cargo install nvidia-checker

打印环境

nvidia-checker

检查期望的环境

nvidia-checker --diff ./config/test.toml
  • 输出示例
17:00:39 [INFO] ===== Start nvidia-checker =====
17:00:39 [INFO] command: uname -r

5.15.0-58-generic

17:00:39 [INFO] command: cat /etc/os-release

PRETTY_NAME="Ubuntu 22.04.1 LTS"
NAME="Ubuntu"
VERSION_ID="22.04"
VERSION="22.04.1 LTS (Jammy Jellyfish)"
VERSION_CODENAME=jammy
ID=ubuntu
ID_LIKE=debian
HOME_URL="https://www.ubuntu.com/"
SUPPORT_URL="https://help.ubuntu.com/"
BUG_REPORT_URL="https://bugs.launchpad.net/ubuntu/"
PRIVACY_POLICY_URL="https://www.ubuntu.com/legal/terms-and-policies/privacy-policy"
UBUNTU_CODENAME=jammy

17:00:39 [INFO] command: cat /proc/driver/nvidia/version

NVRM version: NVIDIA UNIX x86_64 Kernel Module  545.23.08  Mon Nov  6 23:49:37 UTC 2023
GCC version:  gcc version 11.4.0 (Ubuntu 11.4.0-1ubuntu1~22.04) 

17:00:39 [INFO] command: nvcc -V

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Wed_Nov_22_10:17:15_PST_2023
Cuda compilation tools, release 12.3, V12.3.107
Build cuda_12.3.r12.3/compiler.33567101_0

17:00:39 [INFO] command: dpkg -l | grep cudnn

hi  libcudnn8                                         8.9.5.29-1+cuda12.2                     amd64        cuDNN runtime libraries
hi  libcudnn8-dev                                     8.9.5.29-1+cuda12.2                     amd64        cuDNN development libraries and headers
ii  ros-humble-cudnn-cmake-module                     0.0.1-3jammy.20230112.141335            amd64        Exports a CMake module to find cuDNN.

17:00:39 [INFO] command: dpkg -l | grep TensorRT

ii  libnvinfer-bin                                    8.6.1.6-1+cuda12.0                      amd64        TensorRT binaries
hi  libnvinfer-dev                                    8.6.1.6-1+cuda12.0                      amd64        TensorRT development libraries
ii  libnvinfer-dispatch8                              8.6.1.6-1+cuda12.0                      amd64        TensorRT dispatch runtime library
ii  libnvinfer-headers-dev                            8.6.1.6-1+cuda12.0                      amd64        TensorRT development headers
ii  libnvinfer-headers-plugin-dev                     8.6.1.6-1+cuda12.0                      amd64        TensorRT plugin headers
ii  libnvinfer-lean8                                  8.6.1.6-1+cuda12.0                      amd64        TensorRT lean runtime library
hi  libnvinfer-plugin-dev                             8.6.1.6-1+cuda12.0                      amd64        TensorRT plugin libraries
hi  libnvinfer-plugin8                                8.6.1.6-1+cuda12.0                      amd64        TensorRT plugin libraries
ii  libnvinfer-vc-plugin8                             8.6.1.6-1+cuda12.0                      amd64        TensorRT vc-plugin library
hi  libnvinfer8                                       8.6.1.6-1+cuda12.0                      amd64        TensorRT runtime libraries
hi  libnvonnxparsers-dev                              8.6.1.6-1+cuda12.0                      amd64        TensorRT ONNX libraries
hi  libnvonnxparsers8                                 8.6.1.6-1+cuda12.0                      amd64        TensorRT ONNX libraries
hi  libnvparsers-dev                                  8.6.1.6-1+cuda12.0                      amd64        TensorRT parsers libraries
hi  libnvparsers8                                     8.6.1.6-1+cuda12.0                      amd64        TensorRT parsers libraries
ii  ros-humble-tensorrt-cmake-module                  0.0.3-1jammy.20230112.141218            amd64        Exports a CMake module to find TensorRT.

17:00:39 [INFO] ===== Your environment =====
Environment {
    checked_time: "2024.01.27 02:00:39",
    os: "Ubuntu 22.04.1 LTS",
    kernel: "5.15.0-58-generic",
    nvidia_driver: "545.23.08",
    cuda: "cuda_12.3.r12.3",
    cudnn: "8.9.5.29-1+cuda12.2",
    tensorrt: "8.6.1.6-1+cuda12.0",
}
17:00:39 [INFO] ===== Check environment =====
17:00:39 [INFO] os: OK
17:00:39 [INFO] kernel: OK
17:00:39 [INFO] nvidia driver: OK
17:00:39 [INFO] cuda: OK
17:00:39 [INFO] cudnn: OK
17:00:39 [INFO] tensorrt: OK

检查与最新检查的更改

  • 如果您想像自动的 apt update 一样检查NVIDIA软件环境的更改,可以使用以下nvidia-checker
    • 最后运行的nvidia-checker的环境文件位于$HOME/.local/nvidia-checker/latest.toml
nvidia-checker --latest

从源文件运行

  • 克隆仓库
git clone https://github.com/scepter914/nvidia-checker-rs.git
  • 运行
cargo run --release --bin nvidia-checker -- --diff ./config/test.toml

历史

  • 下一个版本
    • 更新README
  • v0.1.3
    • 修复错误
  • v0.1.2
    • 更新cargo.toml
  • v0.1.1
    • 整理包以供库使用
    • 更新README
  • v0.1.0
    • 发布了 nvidia-checker

依赖

~3–12MB
~121K SLoC