5090, windows, can't install

[virtyos]

tried to install ur tool on win 11, got an error after running "pip install -e ."

  Using cached deepspeed-0.18.1.tar.gz (1.6 MB)
  Preparing metadata (setup.py) ... error
  error: subprocess-exited-with-error

  × python setup.py egg_info did not run successfully.
  │ exit code: 1
  ╰─> [9 lines of output]
      Traceback (most recent call last):
        File "<string>", line 2, in <module>
        File "<pip-setuptools-caller>", line 35, in <module>
        File "C:\Users\virty\AppData\Local\Temp\pip-install-atlulw5u\deepspeed_940d54ab7bf44a9496eda6d075ad5efa\setup.py", line 157, in <module>
          assert torch_available, "Unable to pre-compile ops without torch installed. Please install torch before attempting to pre-compile ops."
      AssertionError: Unable to pre-compile ops without torch installed. Please install torch before attempting to pre-compile ops.
      [WARNING] Unable to import torch, pre-compiling ops will be disabled. Please visit https://pytorch.org/ to see how to properly install torch on your system.
       [WARNING]  unable to import torch, please install it if you want to pre-compile any deepspeed ops.
      DS_BUILD_OPS=1

[virtyos]

Does anyone have instructions on how to install this on Windows 11?

[Granddyser]

Yes, I’m currently testing it on Windows but I’ve already precompiled all the required wheels (XFormers, FlashAttention, DeepSpeed, etc.) and released them.
Once testing is complete, I’ll publish everything.

conda env: python 3.10 cuda1.24 torch2.5.0