Spaces:

optigesr
/

lilholt

Runtime error

lilholt / list_devices.py

Upload 3 files

bd39b6b almost 3 years ago

130 Bytes

	import torch

	devices = [f"cuda:{i} => {torch.cuda.get_device_name(i)}" for i in range(torch.cuda.device_count())]

	print(devices)