CompressAI icon indicating copy to clipboard operation
CompressAI copied to clipboard
verified publisher icon InterDigitalInc

Multi-GPU support is now experimental

Open xuezhongcailian opened this issue 1 year ago • 1 comments

Feature

Does this repository currently support multi-GPU experiments

Motivation

Additional context

xuezhongcailian avatar Dec 30 '23 13:12 xuezhongcailian

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.

fracape avatar Feb 28 '24 08:02 fracape