dee-tulu-train / check.py
Javad Taghia
added check.py for cuda
bdd77bf
raw
history blame contribute delete
45 Bytes
import torch
print(torch.cuda.is_available())