From 681d9b12efa89e3b96319fcb038d824c6313b374 Mon Sep 17 00:00:00 2001 From: flybird11111 <1829166702@qq.com> Date: Fri, 15 Dec 2023 18:16:48 +0800 Subject: [PATCH] [doc] update pytorch version in documents. (#5177) * fix aaa fix fix fix * fix * fix * test ci * fix ci fix * update pytorch version in documents --- README.md | 2 +- docs/README-zh-Hans.md | 2 +- docs/source/en/get_started/installation.md | 2 +- docs/source/zh-Hans/get_started/installation.md | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 04a349337..545500591 100644 --- a/README.md +++ b/README.md @@ -372,7 +372,7 @@ Please visit our [documentation](https://www.colossalai.org/) and [examples](htt ## Installation Requirements: -- PyTorch >= 1.11 (PyTorch 2.x in progress) +- PyTorch >= 1.11 and PyTorch <= 2.1 - Python >= 3.7 - CUDA >= 11.0 - [NVIDIA GPU Compute Capability](https://developer.nvidia.com/cuda-gpus) >= 7.0 (V100/RTX20 and higher) diff --git a/docs/README-zh-Hans.md b/docs/README-zh-Hans.md index c6f15b6d6..7bb4a414b 100644 --- a/docs/README-zh-Hans.md +++ b/docs/README-zh-Hans.md @@ -368,7 +368,7 @@ Colossal-AI 为您提供了一系列并行组件。我们的目标是让您的 环境要求: -- PyTorch >= 1.11 (PyTorch 2.x 正在适配中) +- PyTorch >= 1.11 并且 PyTorch <= 2.1 - Python >= 3.7 - CUDA >= 11.0 - [NVIDIA GPU Compute Capability](https://developer.nvidia.com/cuda-gpus) >= 7.0 (V100/RTX20 and higher) diff --git a/docs/source/en/get_started/installation.md b/docs/source/en/get_started/installation.md index 6fc4ce2c9..18607a34c 100644 --- a/docs/source/en/get_started/installation.md +++ b/docs/source/en/get_started/installation.md @@ -1,7 +1,7 @@ # Setup Requirements: -- PyTorch >= 1.11 (PyTorch 2.x in progress) +- PyTorch >= 1.11 and PyTorch <= 2.1 - Python >= 3.7 - CUDA >= 11.0 - [NVIDIA GPU Compute Capability](https://developer.nvidia.com/cuda-gpus) >= 7.0 (V100/RTX20 and higher) diff --git a/docs/source/zh-Hans/get_started/installation.md b/docs/source/zh-Hans/get_started/installation.md index a6c88672b..e75e42530 100755 --- a/docs/source/zh-Hans/get_started/installation.md +++ b/docs/source/zh-Hans/get_started/installation.md @@ -2,7 +2,7 @@ 环境要求: -- PyTorch >= 1.11 (PyTorch 2.x 正在适配中) +- PyTorch >= 1.11 并且 PyTorch <= 2.1 - Python >= 3.7 - CUDA >= 11.0 - [NVIDIA GPU Compute Capability](https://developer.nvidia.com/cuda-gpus) >= 7.0 (V100/RTX20 and higher)