CUDA环境变量 Linux

1. 单版本

CUDA 默认会被安装到 /usr/local/ 目录

安装完成进入此目录会看到如下结构

事实上和驱动不同，同一台机器允许安装多个版本的CUDA，这里暂不讨论

此时我的CUDA目录即为 cuda-12.4 ，绝对路径为:

/usr/local/cuda-12.4

配置环境变量有三种选择，一是配置系统级别的，二是配置用户级别的，三是临时使用

系统级别则修改：

# sudo nano /etc/bash.bashrc
/etc/bash.bashrc

用户级别则修改:

# nano ~/.bashrc
~/.bashrc

根据上面安装的路径和版本，依次添加下面三项，一般只需要替换版本即可，如本文是cuda-12.4

export PATH="/usr/local/cuda-12.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH"
export CUDA_HOME="/usr/local/cuda-12.4"

注意：如果临时使用，依次则依次在命令行窗口中执行上面的三项即可

等修改完成，执行：

source ~/.bashrc #修改那个执行哪条
source /etc/bash.bashrc

2. 多版本选择

function switch_cuda {
    if [[ $1 =~ ^[0-9]+\.[0-9]+$ ]]; then
        if [ -d /usr/local/cuda-$1 ]; then
            sudo rm /usr/local/cuda
            sudo ln -s /usr/local/cuda-$1 /usr/local/cuda
            nvcc --version
        else
            echo "CUDA version $1 not found in /usr/local."
        fi
    else
        echo "Invalid version"
    fi
}