[DAY3]图像分类工具包 MMClassification

代码仓库：https://github.com/open-mmlab/mmclassification

文档教程：https://mmclassification.readthedocs.io/en/latest/

一、安装

1.创建环境

• 加载 anaconda ，创建一个 python 3.8 的环境。

# 创建 python=3.8 的环境
conda create --name mmclassification python=3.8
​
# 激活环境
conda activate mmclassification

• 安装torch

查看已安装的cuda版本

nvcc --version
# Cuda compilation tools, release 11.3, V11.3.58

根据已有的cuda版本安装torch

pytorch下载地址：

https://pytorch.org/get-started/previous-versions/

# CUDA 11.3
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

• 安装 mmcv-full 模块，mmcv-full 模块安装时候需要注意 torch 和 cuda 版本。

参考