每个conda环境设置一个cuda版本

安装cuda

一、确认安装

/usr/local/<cuda-11.7>/bin/nvcc -V  # 更换cuda版本

# cuda安装的默认位置是 /usr/local/ 

成功安装：

二、脚本

conda env list  # 列出所有conda环境

记住，别搞错环境名

        1. 激活

mkdir -p ~/.conda/envs/<环境名称>/etc/conda/activate.d
nano ~/.conda/envs/<环境名称>/etc/conda/activate.d/cuda_env_vars.sh

#################  sh文件内容

#!/bin/sh
export PATH=/usr/local/<cuda版本（文件夹名）>/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/<cuda版本（文件夹名）>/lib64:$LD_LIBRARY_PATH

        2. 关闭

mkdir -p ~/.conda/envs/<环境名称>/etc/conda/deactivate.d
nano ~/.conda/envs/<环境名称>/etc/conda/deactivate.d/cuda_env_vars.sh

#############  sh文件内容

#!/bin/sh
unset PATH
unset LD_LIBRARY_PATH

        3. 重启生效

                conda deactivate

                conda activate env_name