How to use transformers pipeline with multi-gpu?

Viewed 2151
ner_model = pipeline('ner', model=model, tokenizer=tokenizer, device=0, grouped_entities=True)

the device indicated pipeline to use no_gpu=0(only using GPU), please show me how to use multi-gpu.

0 Answers
Related