MMEditng环境搭建

官方文档：mmediting/README.md at master · open-mmlab/mmediting · GitHub

1. 创建&激活环境：

conda create --name MMEditing python=3.8
conda activate MMEditing

2. 安装对应版本Pytorch：

参照官网寻找与自己的环境匹配的版本：

cuda >= 11.6 ：PyTorch

cuda < 11.6 ：Previous PyTorch Versions | PyTorch

eg cuda11.7：

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

eg cuda10.2：

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch

（千万不要图省事跟我装一样的 上官网查对应版本！！）

3. 安装mmcv-full

pip3 install openmim
mim install mmcv-full 
#可指定mmcv-full版本，例如 mim install mmcv-full==1.7.0

4. 安装MMEditing

4.1进入安装位置（默认/home下）

cd /data3/RR

4.2下载MMEditing源码

git clone https://github.com/open-mmlab/mmediting.git

4.3安装MMEditing

cd /data3/RR/mmediting  #上一步的安装位置+‘/mmediting’
pip3 install -e .

5. 检查是否安装成功

可以在命令行打开iPython 这里我就直接在pycharm中远程打开MMEditing 再新建Test.py了

import mmedit
print(mmedit.__version__)

输出版本号，安装成功

顺便检查一下pytorch cuda是否安装正确

import torch
print(torch.cuda.is_available())