search

bowang-lab / Graph-Mamba

Graph-Mamba: Towards Long-Range Graph Sequence Modelling with Selective State Spaces
246 stars 29 forks source link

deepspeed version? #4

Open Maxcharm opened 6 months ago

Maxcharm commented 6 months ago

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 4 months ago

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

BatchClayderman commented 4 months ago

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