不同conda 不同cuda版本方法

RuiyChen2025-06-16 14:36

安装不同的cuda给不同的conda环境使用方法：

第一步：确认系统里有哪几个 CUDA 版本

运行下面命令，看看系统里装了哪些 CUDA：

bash 复制代码

ls /usr/local | grep cuda

你可能会看到类似：

复制代码

第二步：找当前conda环境

bash 复制代码

which python

把找到的环境替换下面的地址

比如输出的是

/home/ruiy/sdb/anaconda3/anaconda3/envs/w/bin/python

那么下面就将

~/anaconda3/envs/gp/etc/conda/activate.d

改为

/home/ruiy/sdb/anaconda3/anaconda3/envs/w/etc/conda/activate.d

第三步：写环境激活替换cuda脚本

bash 复制代码

# 找到当前环境
mkdir -p ~/anaconda3/envs/gp/etc/conda/activate.d
# 
cat <<EOF > ~/anaconda3/envs/gp/etc/conda/activate.d/env_vars.sh
# 指定cuda版本
export CUDA_HOME=/usr/local/cuda-11.8
export PATH=\$CUDA_HOME/bin:\$PATH
export LD_LIBRARY_PATH=\$CUDA_HOME/lib64:\$LD_LIBRARY_PATH
EOF