🐛 Describe the bug
colossalai/cli/launcher/run.py line 157, in get_launch_command
assert torch_version.major == 1
If ColossalAI already supports torch2.0 then we should modify the code.
Environment
Conda installed torch 2.0 by default as the version is not specified in requirements.txt in the tutorial examples.
🐛 Describe the bug
colossalai/cli/launcher/run.py line 157, in get_launch_command
assert torch_version.major == 1
If ColossalAI already supports torch2.0 then we should modify the code.
Environment
Conda installed torch 2.0 by default as the version is not specified in requirements.txt in the tutorial examples.