不需要从零开始装 cuda,各种包。。。

装 cuda

打开 terminal 发现 usr 的最前面没有 (base),表明没有安装 cuda

cuda 是无法 conda install xx 这样安装的,一种方法是从官网上下 torch 和 torchvision

另一种是引用已经安装好的 cuda 环境

在 .bashrc (shell 配置文件) 里加入

# >>> conda initialize >>>
# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/data/xxxx/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"   # conda 的安装路径
if [ $? -eq 0 ]; then
    eval "$__conda_setup"
else
    if [ -f "/data/xxxx/anaconda3/etc/profile.d/conda.sh" ]; then
        . "/data/xxxx/anaconda3/etc/profile.d/conda.sh"
    else
        export PATH="/data/xxxx/anaconda3/bin:$PATH"
    fi
fi
unset __conda_setup
# <<< conda initialize <<<

# <<< cuda init <<<
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-11.1
# <<< cuda init <<<

再输入 source .bashrc 重新应用配置文件即可

拷贝环境

cp -r /data/xxx/env/* /path/to/new/env/