1、创建虚拟环境并激活。
conda create --name openmmlab python=3.8 -y
conda activate openmmlab
2、安装Pytorch。Pytorch官网为:Start Locally | PyTorch。
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
3、使用 MIM安装 MMEngine 和 MMCV。
pip install -U openmim
mim install mmengine
mim install 'mmcv>=2.0.0rc1'
4、安装mmselfsup,注意最后一条语句有个 . 点。
git clone https://github.com/open-mmlab/mmselfsup.git
如果输入上述代码显示网络连接错误或者下载速度特别慢,也可以从github把mmselfsup的压缩包下载下来,然后解压了。然后cd进入下载的路径。
cd mmselfsup
git checkout 1.x
pip install -v -e .
5、测试mmselfsup是否安装成功。
import mmselfsup
print(mmselfsup.__version__)