Skip to content

Commit 2c4516c

Browse files
authored
Add the supported pytorch versions in README (#1512)
1 parent 447d3bb commit 2c4516c

File tree

2 files changed

+22
-4
lines changed

2 files changed

+22
-4
lines changed

README.md

+11-2
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,9 @@
1919
<div>&nbsp;</div>
2020

2121
[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/mmengine)](https://pypi.org/project/mmengine/)
22+
[![pytorch](https://img.shields.io/badge/pytorch-1.6~2.1-yellow)](#installation)
2223
[![PyPI](https://img.shields.io/pypi/v/mmengine)](https://pypi.org/project/mmengine)
2324
[![license](https://img.shields.io/github/license/open-mmlab/mmengine.svg)](https://github.com/open-mmlab/mmengine/blob/main/LICENSE)
24-
[![open issues](https://isitmaintained.com/badge/open/open-mmlab/mmengine.svg)](https://github.com/open-mmlab/mmengine/issues)
25-
[![issue resolution](https://isitmaintained.com/badge/resolution/open-mmlab/mmengine.svg)](https://github.com/open-mmlab/mmengine/issues)
2625

2726
[Introduction](#introduction) |
2827
[Installation](#installation) |
@@ -97,6 +96,16 @@ MMEngine is a foundational library for training deep learning models based on Py
9796

9897
## Installation
9998

99+
<details>
100+
<summary>Supported PyTorch Versions</summary>
101+
102+
| MMEngine | PyTorch | Python |
103+
| ------------------ | ------------ | -------------- |
104+
| main | >=1.6 \<=2.1 | >=3.8, \<=3.11 |
105+
| >=0.9.0, \<=0.10.3 | >=1.6 \<=2.1 | >=3.8, \<=3.11 |
106+
107+
</details>
108+
100109
Before installing MMEngine, please ensure that PyTorch has been successfully installed following the [official guide](https://pytorch.org/get-started/locally/).
101110

102111
Install MMEngine

README_zh-CN.md

+11-2
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,9 @@
1919
<div>&nbsp;</div>
2020

2121
[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/mmengine)](https://pypi.org/project/mmengine/)
22+
[![pytorch](https://img.shields.io/badge/pytorch-1.6~2.1-yellow)](#安装)
2223
[![PyPI](https://img.shields.io/pypi/v/mmengine)](https://pypi.org/project/mmengine)
2324
[![license](https://img.shields.io/github/license/open-mmlab/mmengine.svg)](https://github.com/open-mmlab/mmengine/blob/main/LICENSE)
24-
[![open issues](https://isitmaintained.com/badge/open/open-mmlab/mmengine.svg)](https://github.com/open-mmlab/mmengine/issues)
25-
[![issue resolution](https://isitmaintained.com/badge/resolution/open-mmlab/mmengine.svg)](https://github.com/open-mmlab/mmengine/issues)
2625

2726
[简介](#简介) |
2827
[安装](#安装) |
@@ -102,6 +101,16 @@ MMEngine 是一个基于 PyTorch 实现的,用于训练深度学习模型的
102101

103102
## 安装
104103

104+
<details>
105+
<summary>支持的 PyTorch 版本</summary>
106+
107+
| MMEngine | PyTorch | Python |
108+
| ------------------ | ------------ | -------------- |
109+
| main | >=1.6 \<=2.1 | >=3.8, \<=3.11 |
110+
| >=0.9.0, \<=0.10.3 | >=1.6 \<=2.1 | >=3.8, \<=3.11 |
111+
112+
</details>
113+
105114
在安装 MMEngine 之前,请确保 PyTorch 已成功安装在环境中,可以参考 [PyTorch 官方安装文档](https://pytorch.org/get-started/locally/)
106115

107116
安装 MMEngine

0 commit comments

Comments
 (0)