1. Anaconda安装
- anaconda安装下载
anaconda下载
bash Anaconda3-2023.07-2-Linux-x86_64.sh
- 编辑环境变量配置文件
nano ~/.bashrc
# <<< conda initialize <<<
# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/你自己的目录/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
eval "$__conda_setup"
else
if [ -f "/你自己的目录/anaconda3/etc/profile.d/conda.sh" ]; then
. "/你自己的目录/anaconda3/etc/profile.d/conda.sh"
else
export PATH="/你自己的目录/anaconda3/bin:$PATH"
fi
fi
unset __conda_setup
# <<< conda initialize <<<
# 输入以下命令或者重启系统
source ~/.bashrc
- 创建和管理新环境
创建新环境,执行以下指令创建一个名为 yolov5的新环境:
conda create --name yolov5 python==3.8
执行以下指令激活新环境:
conda activate python yolov5
前面出现括号显示激活成功
其他操作:
离开环境
conda deactivate
删除环境:
conda remove --name yolov5 --all
列出所有环境,可以使用以下指令列出已安装的所有环境:
conda info --envs
复制环境,执行以下指令来创建与现有环境相同的新环境:
conda create --name new_env --clone existing_env
更新环境中的包,可以使用以下指令更新环境中的包:
conda update my_env
2. Pytorch 安装
以管理员身份进入cmd,输入activate yolov5 进入环境目录下,输入安装命令,选择对应版本进行安装
Linux and Windows
# CUDA 10.2
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch
# CUDA 11.3
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
# CUDA 11.6
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge
# CPU Only
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cpuonly -c pytorch
验证是否安装成功
import torch
print(torch.cuda.is_available())
print(torch.__version__)
出现下面结果即为安装成功
3.pycharm配置对应的conda环境
参考博客:https://blog.csdn.net/qq_41737845/article/details/128252041