个人深度学习环境配置

pip配置清华源

-i https://pypi.tuna.tsinghua.edu.cn/simple

创建虚拟环境，安装pytorch

conda create --name XXX python=3.9
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

安装mmcv、mmsegmentation等

pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"
# 通过源码安装mmsegmentation
git clone https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation
pip install -v -e .

验证安装

进入python

# 检查 Pytorch
import torch, torchvision
print('Pytorch 版本', torch.__version__)
print('CUDA 是否可用',torch.cuda.is_available())

# 检查 mmcv
import mmcv
from mmcv.ops import get_compiling_cuda_version, get_compiler_version
print('MMCV版本', mmcv.__version__)
print('CUDA版本', get_compiling_cuda_version())
print('编译器版本', get_compiler_version())

# 检查 mmsegmentation
import mmseg
print('mmsegmentation版本', mmseg.__version__)

其他库安装

将albumentations指向已安装的opencv-python

pip install albumentations --no-binary qudida,albumentations -i https://pypi.tuna.tsinghua.edu.cn/simple