ColossalAI check failed

[forestbat]

This is my report in WSL:

forestbat@DESKTOP-KFFIMMS:~$ pip install colossalai
Defaulting to user installation because normal site-packages is not writeable
WARNING: Ignoring invalid distribution -orch (/home/forestbat/.local/lib/python3.10/site-packages)
WARNING: Ignoring invalid distribution -orch (/home/forestbat/.local/lib/python3.10/site-packages)
Collecting colossalai
  Using cached colossalai-0.2.8.tar.gz (728 kB)
  Preparing metadata (setup.py) ... done
Requirement already satisfied: click in ./.local/lib/python3.10/site-packages (from colossalai) (8.1.3)
Requirement already satisfied: contexttimer in ./.local/lib/python3.10/site-packages (from colossalai) (0.3.3)
Collecting fabric
  Using cached fabric-3.0.0-py3-none-any.whl (53 kB)
Requirement already satisfied: ninja in ./.local/lib/python3.10/site-packages (from colossalai) (1.11.1)
Requirement already satisfied: numpy in ./.local/lib/python3.10/site-packages (from colossalai) (1.24.1)
Requirement already satisfied: packaging in ./.local/lib/python3.10/site-packages (from colossalai) (23.0)
Collecting pre-commit
  Using cached pre_commit-3.2.1-py2.py3-none-any.whl (202 kB)
Requirement already satisfied: psutil in ./.local/lib/python3.10/site-packages (from colossalai) (5.9.4)
Collecting rich
  Using cached rich-13.3.3-py3-none-any.whl (238 kB)
Requirement already satisfied: torch<2.0,>=1.11 in ./.local/lib/python3.10/site-packages (from colossalai) (1.13.1)
Requirement already satisfied: tqdm in ./.local/lib/python3.10/site-packages (from colossalai) (4.65.0)
Requirement already satisfied: typing-extensions in ./.local/lib/python3.10/site-packages (from torch<2.0,>=1.11->colossalai) (4.4.0)
Requirement already satisfied: nvidia-cuda-nvrtc-cu11==11.7.99 in ./.local/lib/python3.10/site-packages (from torch<2.0,>=1.11->colossalai) (11.7.99)
Requirement already satisfied: nvidia-cuda-runtime-cu11==11.7.99 in ./.local/lib/python3.10/site-packages (from torch<2.0,>=1.11->colossalai) (11.7.99)
Requirement already satisfied: nvidia-cublas-cu11==11.10.3.66 in ./.local/lib/python3.10/site-packages (from torch<2.0,>=1.11->colossalai) (11.10.3.66)
Requirement already satisfied: nvidia-cudnn-cu11==8.5.0.96 in ./.local/lib/python3.10/site-packages (from torch<2.0,>=1.11->colossalai) (8.5.0.96)
Requirement already satisfied: setuptools in /usr/lib/python3/dist-packages (from nvidia-cublas-cu11==11.10.3.66->torch<2.0,>=1.11->colossalai) (59.6.0)
Requirement already satisfied: wheel in /usr/lib/python3/dist-packages (from nvidia-cublas-cu11==11.10.3.66->torch<2.0,>=1.11->colossalai) (0.37.1)
Requirement already satisfied: invoke>=2.0 in ./.local/lib/python3.10/site-packages (from fabric->colossalai) (2.0.0)
Collecting paramiko>=2.4
  Using cached paramiko-3.1.0-py3-none-any.whl (211 kB)
Requirement already satisfied: virtualenv>=20.10.0 in ./.local/lib/python3.10/site-packages (from pre-commit->colossalai) (20.21.0)
Requirement already satisfied: nodeenv>=0.11.1 in ./.local/lib/python3.10/site-packages (from pre-commit->colossalai) (1.7.0)
Requirement already satisfied: pyyaml>=5.1 in /usr/lib/python3/dist-packages (from pre-commit->colossalai) (5.4.1)
Requirement already satisfied: cfgv>=2.0.0 in ./.local/lib/python3.10/site-packages (from pre-commit->colossalai) (3.3.1)
Requirement already satisfied: identify>=1.0.0 in ./.local/lib/python3.10/site-packages (from pre-commit->colossalai) (2.5.22)
Requirement already satisfied: markdown-it-py<3.0.0,>=2.2.0 in ./.local/lib/python3.10/site-packages (from rich->colossalai) (2.2.0)
Requirement already satisfied: pygments<3.0.0,>=2.13.0 in ./.local/lib/python3.10/site-packages (from rich->colossalai) (2.14.0)
Requirement already satisfied: mdurl~=0.1 in ./.local/lib/python3.10/site-packages (from markdown-it-py<3.0.0,>=2.2.0->rich->colossalai) (0.1.2)
Requirement already satisfied: cryptography>=3.3 in /usr/lib/python3/dist-packages (from paramiko>=2.4->fabric->colossalai) (3.4.8)
Collecting pynacl>=1.5
  Using cached PyNaCl-1.5.0-cp36-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.manylinux_2_24_x86_64.whl (856 kB)
Requirement already satisfied: bcrypt>=3.2 in ./.local/lib/python3.10/site-packages (from paramiko>=2.4->fabric->colossalai) (4.0.1)
Requirement already satisfied: platformdirs<4,>=2.4 in ./.local/lib/python3.10/site-packages (from virtualenv>=20.10.0->pre-commit->colossalai) (3.2.0)
Requirement already satisfied: filelock<4,>=3.4.1 in ./.local/lib/python3.10/site-packages (from virtualenv>=20.10.0->pre-commit->colossalai) (3.9.0)
Requirement already satisfied: distlib<1,>=0.3.6 in ./.local/lib/python3.10/site-packages (from virtualenv>=20.10.0->pre-commit->colossalai) (0.3.6)
Requirement already satisfied: cffi>=1.4.1 in ./.local/lib/python3.10/site-packages (from pynacl>=1.5->paramiko>=2.4->fabric->colossalai) (1.15.1)
Requirement already satisfied: pycparser in ./.local/lib/python3.10/site-packages (from cffi>=1.4.1->pynacl>=1.5->paramiko>=2.4->fabric->colossalai) (2.21)
Building wheels for collected packages: colossalai
  Building wheel for colossalai (setup.py) ... done
  Created wheel for colossalai: filename=colossalai-0.2.8-py3-none-any.whl size=930944 sha256=a6e108e941d4043b7a75f634d72cfe01904c128d87aadeb4072bf71df40bd485
  Stored in directory: /home/forestbat/.cache/pip/wheels/91/4d/77/9e915ab39a58f922cb9c4db126e8a082b56131e81339d64fcb
Successfully built colossalai
WARNING: Ignoring invalid distribution -orch (/home/forestbat/.local/lib/python3.10/site-packages)
Installing collected packages: rich, pynacl, pre-commit, paramiko, fabric, colossalai
WARNING: Ignoring invalid distribution -orch (/home/forestbat/.local/lib/python3.10/site-packages)
WARNING: Ignoring invalid distribution -orch (/home/forestbat/.local/lib/python3.10/site-packages)
WARNING: Ignoring invalid distribution -orch (/home/forestbat/.local/lib/python3.10/site-packages)
WARNING: Ignoring invalid distribution -orch (/home/forestbat/.local/lib/python3.10/site-packages)
WARNING: Ignoring invalid distribution -orch (/home/forestbat/.local/lib/python3.10/site-packages)
WARNING: Ignoring invalid distribution -orch (/home/forestbat/.local/lib/python3.10/site-packages)
Successfully installed colossalai-0.2.8 fabric-3.0.0 paramiko-3.1.0 pre-commit-3.2.1 pynacl-1.5.0 rich-13.3.3


forestbat@DESKTOP-KFFIMMS:~$ colossalai check -i
Traceback (most recent call last):
  File "/home/forestbat/.local/bin/colossalai", line 5, in <module>
    from colossalai.cli import cli
  File "/home/forestbat/.local/lib/python3.10/site-packages/colossalai/__init__.py", line 1, in <module>
    from .initialize import (
  File "/home/forestbat/.local/lib/python3.10/site-packages/colossalai/initialize.py", line 11, in <module>
    import torch.nn as nn
  File "/home/forestbat/.local/lib/python3.10/site-packages/torch/nn/__init__.py", line 1, in <module>
    from .modules import *  # noqa: F403
  File "/home/forestbat/.local/lib/python3.10/site-packages/torch/nn/modules/__init__.py", line 1, in <module>
    from .module import Module
  File "/home/forestbat/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 8, in <module>
    from ..parameter import Parameter
  File "/home/forestbat/.local/lib/python3.10/site-packages/torch/nn/parameter.py", line 2, in <module>
    from torch._C import _disabled_torch_function_impl
ModuleNotFoundError: No module named 'torch._C'

So what happened to this program?

[binmakeswell]

Hi @forestbat Have you installed PyTorch yet? What's your PyTorch version?
Please check requirements https://github.com/hpcaitech/ColossalAI#Installation
Thanks.