如何在多GPU中使用变压器管道?

时间:2020-10-04 09:34:37

标签: python huggingface-transformers

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

设备 指示管道使用no_gpu = 0(仅使用GPU),请教我如何使用multi-gpu。

0 个答案:

没有答案