semmyKG / troubleshoot.py
semmyk's picture
first commit
c7ce07f
import torch
torch.cuda.is_available()
torch_staus = torch.cuda.is_available()
print(f'torch available: {torch_staus}')