ai-voice-cloning-fork/src/list_devices.py

5 lines
134 B
Python
Raw Normal View History

2023-02-17 00:08:27 +00:00
import torch
devices = [f"cuda:{i} => {torch.cuda.get_device_name(i)}" for i in range(torch.cuda.device_count())]
print(devices)