Cuda device instead of use_gpu=True when multiple GPUs are available

[mrgransky]

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?

Cheers,