上海制造网站公司西安在线

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

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

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

ls /usr/local | grep cuda

你可能会看到类似：

第二步：找当前conda环境

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脚本

# 找到当前环境
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