xxie commited on
Commit
6b181f4
1 Parent(s): a26c57b

check pyt3d version

Browse files
Files changed (1) hide show
  1. test.py +12 -0
test.py CHANGED
@@ -13,6 +13,18 @@ if __name__ == '__main__':
13
  print("CUDA_PATH:", os.getenv("CUDA_PATH"))
14
  print("CUDA available?", torch.cuda.is_available())
15
  demo = gr.Blocks(title="HDM Interaction Reconstruction Demo")
 
 
 
 
 
 
 
 
 
 
 
 
16
 
17
  with demo:
18
  input_rgb = gr.Image(label='Input RGB', type='numpy')
 
13
  print("CUDA_PATH:", os.getenv("CUDA_PATH"))
14
  print("CUDA available?", torch.cuda.is_available())
15
  demo = gr.Blocks(title="HDM Interaction Reconstruction Demo")
16
+ # get the url to install pytorch3d
17
+ import sys
18
+ import torch
19
+
20
+ pyt_version_str = torch.__version__.split("+")[0].replace(".", "")
21
+ version_str = "".join([
22
+ f"py3{sys.version_info.minor}_cu",
23
+ torch.version.cuda.replace(".", ""),
24
+ f"_pyt{pyt_version_str}"
25
+ ])
26
+ install = f'https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html'
27
+ print("Install cmd:", install)
28
 
29
  with demo:
30
  input_rgb = gr.Image(label='Input RGB', type='numpy')