From 2011b1356a977c43bfa9cc9fcddc12089932765e Mon Sep 17 00:00:00 2001 From: binmakeswell Date: Thu, 16 May 2024 13:54:32 +0800 Subject: [PATCH 1/2] [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) From e57812c6727e325971cb0d8769c0789c088f62ae Mon Sep 17 00:00:00 2001 From: botbw Date: Fri, 17 May 2024 13:42:18 +0800 Subject: [PATCH 2/2] [chore] Update placement_policy.py --- colossalai/zero/gemini/placement_policy.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/colossalai/zero/gemini/placement_policy.py b/colossalai/zero/gemini/placement_policy.py index c0f92fa50..e9e871b46 100644 --- a/colossalai/zero/gemini/placement_policy.py +++ b/colossalai/zero/gemini/placement_policy.py @@ -45,11 +45,6 @@ class PlacementPolicy(ABC): raise NotImplementedError -import os - -rank = int(os.environ["RANK"]) - - class StaticPlacementPolicy(PlacementPolicy): def __init__( self,