Linux CUDA环境变量详解：版本配置及修改指南

本文将详细介绍Linux系统中常见的CUDA环境变量及其含义，并以CUDA 11.8版本为例，演示如何配置和修改这些环境变量。

常见CUDA环境变量

在Linux系统中，CUDA及其相关库的版本和路径通常由一系列环境变量控制。以下是部分常见环境变量及其含义：

NV_NVML_DEV_VERSION: NVIDIA Management Library (NVML) 的开发版本号。- NV_LIBNCCL_DEV_PACKAGE: NVIDIA Collective Communications Library (NCCL) 的开发包版本号。- NVIDIA_REQUIRE_CUDA: 指定所需的CUDA版本和NVIDIA GPU驱动版本。- NV_LIBCUBLAS_DEV_PACKAGE: NVIDIA CUBLAS 的开发包版本号。- NV_NVTX_VERSION: NVIDIA Tools Extension (NVTX) 的版本号。- NV_CUDA_CUDART_DEV_VERSION: CUDA Runtime (CUDART) 的开发版本号。- NV_LIBCUSPARSE_VERSION: NVIDIA CUSPARSE 的版本号。- NV_LIBNPP_VERSION: NVIDIA Performance Primitives (NPP) 的版本号。- NV_CUDNN_PACKAGE: CUDA Deep Neural Network (cuDNN) 的包版本号。- NV_NVPROF_DEV_PACKAGE: NVIDIA Profiler (nvprof) 的开发包版本号。- NV_LIBNPP_PACKAGE: NVIDIA Performance Primitives (NPP) 的包版本号。- NV_LIBCUBLAS_DEV_VERSION: NVIDIA CUBLAS 的开发版本号。- NV_LIBCUBLAS_DEV_PACKAGE_NAME: NVIDIA CUBLAS 的开发包名称。- NV_CUDA_CUDART_VERSION: CUDA Runtime (CUDART) 的版本号。- CUDA_VERSION: CUDA 的版本号。- SINGULARITY_CONTAINER: Singularity 容器的路径。- NV_LIBCUBLAS_PACKAGE: NVIDIA CUBLAS 的包版本号。- NV_LIBNPP_DEV_PACKAGE: NVIDIA Performance Primitives (NPP) 的开发包版本号。- NV_LIBCUBLAS_PACKAGE_NAME: NVIDIA CUBLAS 的包名称。- NV_LIBNPP_DEV_VERSION: NVIDIA Performance Primitives (NPP) 的开发版本号。- NV_LIBCUSPARSE_DEV_VERSION: NVIDIA CUSPARSE 的开发版本号。- NV_CUDA_LIB_VERSION: CUDA 库的版本号。- NV_CUDNN_PACKAGE_DEV: CUDA Deep Neural Network (cuDNN) 的开发包版本号。- NV_CUDA_COMPAT_PACKAGE: CUDA 兼容性包的版本号。- NV_LIBNCCL_PACKAGE: NVIDIA Collective Communications Library (NCCL) 的包版本号。- NV_NVPROF_VERSION: NVIDIA Profiler (nvprof) 的版本号。

CUDA 11.8 版本配置示例

假设您已安装CUDA 11.8，并希望配置相关环境变量，以下是部分关键环境变量的配置示例：bashexport CUDA_VERSION=11.8.0export NV_LIBCUBLAS_DEV_PACKAGE='libcublas-dev-11-8'export NV_LIBCUBLAS_PACKAGE='libcublas-11-8'export NV_LIBNCCL_DEV_PACKAGE='libnccl-dev=2.14.3-1+cuda11.8'export NV_LIBNCCL_PACKAGE='libnccl2=2.14.3-1+cuda11.8'export NV_CUDNN_PACKAGE='libcudnn8=8.6.0.163-1+cuda11.8'export NV_CUDNN_PACKAGE_DEV='libcudnn8-dev=8.6.0.163-1+cuda11.8'

请注意，以上只是一些示例，实际配置可能因系统环境和软件版本而异。

修改CUDA环境变量

您可以通过以下两种方式修改CUDA环境变量：

编辑 bash 配置文件: 您可以编辑 ~/.bashrc 或 ~/.bash_profile 文件，并将环境变量的设置添加到文件末尾。修改完成后，使用 source ~/.bashrc 或 source ~/.bash_profile 命令使更改生效。
使用 export 命令: 您可以在终端中直接使用 export 命令设置环境变量。例如，要将 CUDA_VERSION 设置为 11.8.0，可以使用以下命令：

bash export CUDA_VERSION=11.8.0

这种方式设置的环境变量只在当前终端会话有效，关闭终端后会失效。

希望本文能帮助您更好地理解和配置Linux系统下的CUDA环境变量。