cuda12.1环境变量设置
时间: 2024-08-21 21:02:53 浏览: 172
linux下cuda fortran环境搭建
CUDA (Compute Unified Device Architecture) 环境变量设置是为了让编程语言如 C++ 或 Python 能够识别和访问 NVIDIA GPU 的 CUDA 功能。对于 CUDA 12.1,主要需要配置以下几个环境变量:
1. **CUDA_HOME**: 这个变量指向 CUDA 安装目录。例如,如果你安装了 CUDA 12.1 到默认路径 `/usr/local/cuda/12.1`,则设置为 `export CUDA_HOME=/usr/local/cuda/12.1`。
2. **PATH**: 需要在系统路径中添加 CUDA 工具链的 bin 目录,使得命令行工具能够被找到。添加类似这样的路径:`export PATH=${CUDA_HOME}/bin:$PATH`。
3. **LD_LIBRARY_PATH** 或者 **DYLD_LIBRARY_PATH** (取决于操作系统): 这里指定 CUDA 库文件的位置,如果库不在默认位置,可以设置为`${CUDA_HOME}/lib64`。
4. **CUDA_CUDART_LIBRARY** 和 **CUDA_CUDA煨** (如果有使用 cuDNN): 如果你使用的应用程序链接到了特定版本的 cuDNN,也需要设置相应的库路径。
5. **CUDA_NVSIM**: 如果你使用 NVidia Visual Profiler (nvvp),还需要设置这个环境变量指向 nvvp 可执行文件的目录。
完成设置后,记得运行 `source ~/.bashrc` 或重启终端使环境变量生效。
阅读全文