Multi-GPU support is now experimental

InterDigitalInc / CompressAI

A PyTorch library and evaluation platform for end-to-end compression research

https://interdigitalinc.github.io/CompressAI/

BSD 3-Clause Clear License

1.21k stars 232 forks source link

Multi-GPU support is now experimental #268

Open xuezhongcailian opened 10 months ago

xuezhongcailian commented 10 months ago

Feature

Does this repository currently support multi-GPU experiments

Motivation

Additional context

fracape commented 8 months ago

Hi, Not at the moment. For reference, there is a PR on that (#223) which requires some cleaning and a more robust / generic way to gather gpu info than using WORLD_SIZE directly.