PyTorch开源图像分类工具MMClassification如何详细解析?
- 内容介绍
- 文章标签
- 相关推荐
本文共计572个文字,预计阅读时间需要3分钟。
MMClassification是一个基于PyTorch的开放源代码图像分类工具箱,属于OpenMMLab项目的一部分。源码托管在GitHub上,最新发布版本为v0.23.2,遵循Apache-2.0许可协议。它支持在Windows、Linux和Mac上运行。
+ 1. 安装:使用命令`pip install mmclassification`进行安装。
MMClassification是一个基于PyTorch的开源图像分类工具箱,是OpenMMLab项目的一部分,源码传送门,最新发布版本为v0.23.2,License为Apache-2.0。它支持在Windows、Linux和Mac上运行。
1.安装:使用conda安装
(1).创建openmmlab虚拟环境:
conda create -n openmmlab python=3.8
conda activate openmmlab
(2).安装PyTorch:这里PyTorch使用1.11.0版本,CUDA使用10.2版本,此CUDA版本对PyTorch各版本都支持
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c pytorch
(3).安装MMCV:MMCV有两个版本,这里安装带CUDA的mmcv-full
1).mmcv-full: 完整版,包含所有的特性以及丰富的开箱即用的CUDA算子,安装此版本需要较长时间。
本文共计572个文字,预计阅读时间需要3分钟。
MMClassification是一个基于PyTorch的开放源代码图像分类工具箱,属于OpenMMLab项目的一部分。源码托管在GitHub上,最新发布版本为v0.23.2,遵循Apache-2.0许可协议。它支持在Windows、Linux和Mac上运行。
+ 1. 安装:使用命令`pip install mmclassification`进行安装。
MMClassification是一个基于PyTorch的开源图像分类工具箱,是OpenMMLab项目的一部分,源码传送门,最新发布版本为v0.23.2,License为Apache-2.0。它支持在Windows、Linux和Mac上运行。
1.安装:使用conda安装
(1).创建openmmlab虚拟环境:
conda create -n openmmlab python=3.8
conda activate openmmlab
(2).安装PyTorch:这里PyTorch使用1.11.0版本,CUDA使用10.2版本,此CUDA版本对PyTorch各版本都支持
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c pytorch
(3).安装MMCV:MMCV有两个版本,这里安装带CUDA的mmcv-full
1).mmcv-full: 完整版,包含所有的特性以及丰富的开箱即用的CUDA算子,安装此版本需要较长时间。

