oneScotch commited on
Commit
c034d58
1 Parent(s): 042f34b

move to debian

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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/.pyenv/versions/3.9.18/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
 
 
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