3 lines
147 B
Python
3 lines
147 B
Python
import torch
|
|
print(f"number of GPUs: {torch.cuda.device_count()}")
|
|
print([torch.cuda.get_device_name(i) for i in range(torch.cuda.device_count())]) |