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)