PyTorch 环境配置
一个很快的方法
不需要从零开始装 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/