Modify torch version requirement to adapt torch 2.0 (#3896)

1 year ago · 8065cc5fba
parent dbb32692d2
commit 8065cc5fba
1 changed files with 1 additions and 1 deletions
--- 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 = [