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