leobcc
commited on
Commit
·
e348bce
1
Parent(s):
1fc711c
Add files to check installs
Browse files- check_gpu_availability +8 -0
- 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__)
|