How to turn on or off gpu

[GoldDRoge]

I want to control when to turn on or off gpu but i cant find how to do that. Can u help me

[hbredin]

pyannote pipelines will always use GPU when there is one available.

You could set CUDA_VISIBLE_DEVICES environment variable to hide GPUs:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = ""

from pyannote.audio import Pipeline
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
pipeline("audio.wav") # will not use GPU

[hbredin]

This PR adds a new pipeline.to(device) API. Would you mind testing it?

[GoldDRoge]

thanks for your help. its work