yslan commited on
Commit
2a57a54
1 Parent(s): 8999766
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -30,9 +30,10 @@ version_str="".join([
30
  ])
31
  # install pytorch3d with the right version
32
  os.system('pip install iopath')
 
33
  os.system(f'pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html')
34
 
35
- import pytorch3d
36
 
37
  # install 2dgs dependencies
38
  os.system('pip install ./submodules/simple-knn')
 
30
  ])
31
  # install pytorch3d with the right version
32
  os.system('pip install iopath')
33
+ os.system('pip install fvcore')
34
  os.system(f'pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html')
35
 
36
+ import pytorch3d # check whether the version matches
37
 
38
  # install 2dgs dependencies
39
  os.system('pip install ./submodules/simple-knn')
requirements.txt CHANGED
@@ -45,4 +45,5 @@ tritonclient
45
  gradio==4.43.0
46
  rembg
47
  onnxruntime
 
48
  # git+https://github.com/facebookresearch/pytorch3d.git@stable
 
45
  gradio==4.43.0
46
  rembg
47
  onnxruntime
48
+ fvcore
49
  # git+https://github.com/facebookresearch/pytorch3d.git@stable