COMET
COMET copied to clipboard
[QUESTION] how to enable multi-gpu when calling the predict method
The predict method is called by setting gpus=2, but only one gpu is used.
model.predict(samples=samples,batch_size=2, gpus=2, devices=2, accelerator="gpu", strategy="ddp")
you need to specify clear device number like "devices=[1,2]":
model.predict(samples=samples,batch_size=2,_ gpus=2, devices=[1,2], accelerator="gpu", strategy="ddp")
or don‘t pass the 'devices' like:
model.predict(samples=samples,batch_size=2,_ gpus=2, accelerator="gpu", strategy="ddp")