bowang-lab/Graph-Mamba

deepspeed version?

Maxcharm opened this issue · 2 comments

Maxcharm commented

Hi, thank you for the amazing code :)

i was trying the FLOPs profiler and would like to know which version of deepspeed you used for this experiment, since the latest version seems uncompatible with torch 1.13.0

BatchClayderman commented

I tried deepspeed-0.11.2+cuda118-cp310-cp310-win_amd64.whl on Windows 10 and it worked.

BatchClayderman commented

On Ubuntu 22.04, I installed deepspeed==0.14.4 via pip install deepspeed directly. It worked.