From 8065cc5fbac578c9578209cd6768c2242c9aa20a Mon Sep 17 00:00:00 2001
From: Liu Ziming <38985202+MaruyamaAya@users.noreply.github.com>
Date: Mon, 5 Jun 2023 15:57:35 +0800
Subject: [PATCH] Modify torch version requirement to adapt torch 2.0 (#3896)

---
 colossalai/cli/launcher/run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colossalai/cli/launcher/run.py b/colossalai/cli/launcher/run.py
index 027a10aa8..daa5107ca 100644
--- a/colossalai/cli/launcher/run.py
+++ b/colossalai/cli/launcher/run.py
@@ -154,7 +154,7 @@ def get_launch_command(
         extra_launch_args = dict()
 
     torch_version = version.parse(torch.__version__)
-    assert torch_version.major == 1
+    assert torch_version.major >= 1
 
     if torch_version.minor < 9:
         cmd = [