改变cuda和pytorch版本

!pip --default-timeout=1000 install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html

查看torch版本

import torch
print(torch.version)

安装mmcv-full

!pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html

安装mmclassification

克隆并安装mmcls

!git clone https://github.com/open-mmlab/mmclassification.git

切换到目录

%cd mmclassification/

源码安装

!pip install -e .

检查mmcls版本

import mmcls
print(mmcls.version)