安装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