From 2011b1356a977c43bfa9cc9fcddc12089932765e Mon Sep 17 00:00:00 2001 From: binmakeswell Date: Thu, 16 May 2024 13:54:32 +0800 Subject: [PATCH] [misc] Update PyTorch version in docs (#5724) * [misc] Update PyTorch version in docs * [misc] Update PyTorch version in docs --- README.md | 2 +- docs/README-zh-Hans.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 9e215df63..3157d74c9 100644 --- a/README.md +++ b/README.md @@ -418,7 +418,7 @@ Please visit our [documentation](https://www.colossalai.org/) and [examples](htt ## Installation Requirements: -- PyTorch >= 1.11 and PyTorch <= 2.1 +- 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 2e5437752..41110612c 100644 --- a/docs/README-zh-Hans.md +++ b/docs/README-zh-Hans.md @@ -413,7 +413,7 @@ Colossal-AI 为您提供了一系列并行组件。我们的目标是让您的 环境要求: -- PyTorch >= 1.11 并且 PyTorch <= 2.1 +- 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)