📜  设置 cuda 路径 - Shell-Bash (1)

📅  最后修改于: 2023-12-03 14:57:38.409000             🧑  作者: Mango

设置 CUDA 路径 - Shell/Bash

当我们在使用 CUDA 编程时,需要设置 CUDA 的路径,以便编译和链接 CUDA 代码时能够找到所需的库和头文件。

具体地,在 Linux 系统中,在 Shell 或 Bash 环境下,我们可以通过以下几种方式来设置 CUDA 路径:

1. 直接设置环境变量

通过在 Shell/Bash 中输入以下命令,可以将 CUDA 的路径设置为 /usr/local/cuda

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

注意,如果您的 CUDA 安装在另一个目录下,则需要将上面的路径改为您的 CUDA 路径。另外,如果您在之前已经设置过环境变量,则需要将${PATH}${LD_LIBRARY_PATH}替换为您之前设置的环境变量。

2. 在 .bashrc 或 .bash_profile 中添加环境变量

如果您希望每次启动 Shell/Bash 后都能够自动设置 CUDA 路径,可以将以上命令添加到您的 .bashrc.bash_profile 文件中:

echo 'export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc

或者:

echo 'export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}' >> ~/.bash_profile
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bash_profile

这样每次打开一个新的 Shell/Bash 窗口时,这些配置都会生效。

3. 使用 module 命令加载 CUDA 模块

如果您的系统已经安装了 module 命令,并且已经安装了 CUDA 模块,您可以使用以下命令来加载 CUDA 模块:

module load cuda

这将自动将 CUDA 路径添加到您的环境变量中,以便使用 CUDA。

以上是在 Shell/Bash 中设置 CUDA 路径的几种方式。通过这些方法,您可以轻松地将 CUDA 路径设置为您系统中的实际路径,以便编写和运行 CUDA 程序。