Spaces:
Build error
Build error
oneScotch
commited on
Commit
·
c034d58
1
Parent(s):
042f34b
move to debian
Browse files
app.py
CHANGED
@@ -13,17 +13,17 @@ try:
|
|
13 |
except:
|
14 |
os.system('pip install /home/user/app/main/transformer_utils')
|
15 |
|
16 |
-
os.system('cp -rf /home/user/app/assets/conversions.py /home/user/.
|
17 |
DEFAULT_MODEL='smpler_x_h32'
|
18 |
OUT_FOLDER = '/home/user/app/demo_out'
|
19 |
os.makedirs(OUT_FOLDER, exist_ok=True)
|
20 |
num_gpus = 1 if torch.cuda.is_available() else -1
|
21 |
-
print("!!!", torch.cuda.is_available())
|
22 |
-
print(torch.cuda.device_count())
|
23 |
-
print(torch.version.cuda)
|
24 |
-
index = torch.cuda.current_device()
|
25 |
-
print(index)
|
26 |
-
print(torch.cuda.get_device_name(index))
|
27 |
from main.inference import Inferer
|
28 |
inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
|
29 |
|
|
|
13 |
except:
|
14 |
os.system('pip install /home/user/app/main/transformer_utils')
|
15 |
|
16 |
+
os.system('cp -rf /home/user/app/assets/conversions.py /home/user/.local/lib/python3.9/site-packages/torchgeometry/core/conversions.py')
|
17 |
DEFAULT_MODEL='smpler_x_h32'
|
18 |
OUT_FOLDER = '/home/user/app/demo_out'
|
19 |
os.makedirs(OUT_FOLDER, exist_ok=True)
|
20 |
num_gpus = 1 if torch.cuda.is_available() else -1
|
21 |
+
# print("!!!", torch.cuda.is_available())
|
22 |
+
# print(torch.cuda.device_count())
|
23 |
+
# print(torch.version.cuda)
|
24 |
+
# index = torch.cuda.current_device()
|
25 |
+
# print(index)
|
26 |
+
# print(torch.cuda.get_device_name(index))
|
27 |
from main.inference import Inferer
|
28 |
inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
|
29 |
|