Я не могу запустить WHISPER openai на графическом процессоре в vscode ⇐ Python

[Anonymous]

С помощью этого кода я получаю такой результат для model.device = cpu. Как мне переключиться на свой графический процессор (ноутбук msi + nvidia rtx 2060)
import whisper
from IPython.display import Audio

model = whisper.load_model("base")
Audio("audioingles.mp3")

print (model.device)