NPChef / backend /check.py
mehulr910's picture
initial commit
6f9dac7
raw
history blame contribute delete
83 Bytes
import torch
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))