Open mrgransky opened 5 months ago
Given the multilingual pipeline:
import trankit from trankit import Pipeline lemmatizer_multi_lingual_pipeline = Pipeline( lang='finnish-ftb', gpu=True, # <<<<============== using device="cuda:0" ? instead? ==============>>>> embedding='xlm-roberta-large', cache_dir='/path/2/cache_dir', ) lemmatizer_multi_lingual_pipeline.add('english') lemmatizer_multi_lingual_pipeline.add('swedish') lemmatizer_multi_lingual_pipeline.add('danish') lemmatizer_multi_lingual_pipeline.add('russian') lemmatizer_multi_lingual_pipeline.add('french') lemmatizer_multi_lingual_pipeline.add('german') lemmatizer_multi_lingual_pipeline.set_auto(True)
Instead of gpu=True, Is it possible to specify device="cuda:0" or device="cuda:2" when multiple GPUs are available in a machine?
gpu=True
device="cuda:0"
device="cuda:2"
Cheers,
Given the multilingual pipeline:
Instead of
gpu=True
, Is it possible to specifydevice="cuda:0"
ordevice="cuda:2"
when multiple GPUs are available in a machine?Cheers,