Javad Taghia commited on
Commit
bdd77bf
·
1 Parent(s): dbb959c

added check.py for cuda

Browse files
Files changed (1) hide show
  1. check.py +2 -0
check.py ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ import torch
2
+ print(torch.cuda.is_available())