leobcc commited on
Commit
e348bce
·
1 Parent(s): 1fc711c

Add files to check installs

Browse files
Files changed (2) hide show
  1. check_gpu_availability +8 -0
  2. check_kaolin_install +3 -0
check_gpu_availability ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+
3
+ if torch.cuda.is_available():
4
+ device = torch.device("cuda")
5
+ print(f"GPU {torch.cuda.get_device_name(0)} is available")
6
+ else:
7
+ device = torch.device("cpu")
8
+ print("No GPU available, using CPU instead")
check_kaolin_install ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ import kaolin
2
+
3
+ print(kaolin.__version__)