Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
0467139
1
Parent(s):
b1b4917
Add check for CUDA availability in app.py
Browse files
app.py
CHANGED
|
@@ -10,6 +10,7 @@ os.environ['SPCONV_ALGO'] = 'native'
|
|
| 10 |
from typing import *
|
| 11 |
import torch
|
| 12 |
print(torch.__version__)
|
|
|
|
| 13 |
os.system("nvidia-smi")
|
| 14 |
import numpy as np
|
| 15 |
import imageio
|
|
|
|
| 10 |
from typing import *
|
| 11 |
import torch
|
| 12 |
print(torch.__version__)
|
| 13 |
+
print(torch.cuda.is_available())
|
| 14 |
os.system("nvidia-smi")
|
| 15 |
import numpy as np
|
| 16 |
import imageio
|