📅  最后修改于: 2023-12-03 14:57:38.409000             🧑  作者: Mango
当我们在使用 CUDA 编程时,需要设置 CUDA 的路径,以便编译和链接 CUDA 代码时能够找到所需的库和头文件。
具体地,在 Linux 系统中,在 Shell 或 Bash 环境下,我们可以通过以下几种方式来设置 CUDA 路径:
通过在 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}
替换为您之前设置的环境变量。
如果您希望每次启动 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 窗口时,这些配置都会生效。
如果您的系统已经安装了 module 命令,并且已经安装了 CUDA 模块,您可以使用以下命令来加载 CUDA 模块:
module load cuda
这将自动将 CUDA 路径添加到您的环境变量中,以便使用 CUDA。
以上是在 Shell/Bash 中设置 CUDA 路径的几种方式。通过这些方法,您可以轻松地将 CUDA 路径设置为您系统中的实际路径,以便编写和运行 CUDA 程序。