haodongli commited on
Commit
0bb584a
·
1 Parent(s): 971b51c

Update gradio_demo.py

Browse files
Files changed (1) hide show
  1. gradio_demo.py +5 -0
gradio_demo.py CHANGED
@@ -5,8 +5,13 @@ import os
5
  if os.environ.get('SUBMODULES_INSTALLED') != "True":
6
  os.environ['SUBMODULES_INSTALLED'] = "True"
7
  os.environ["TORCH_CUDA_ARCH_LIST"] = "3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX"
 
8
  os.system("python -m pip install git+https://github.com/YixunLiang/simple-knn.git")
 
9
  os.system("python -m pip install git+https://github.com/YixunLiang/diff-gaussian-rasterization.git")
 
 
 
10
 
11
  from train import *
12
 
 
5
  if os.environ.get('SUBMODULES_INSTALLED') != "True":
6
  os.environ['SUBMODULES_INSTALLED'] = "True"
7
  os.environ["TORCH_CUDA_ARCH_LIST"] = "3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX"
8
+ print('==> TORCH_CUDA_ARCH_LIST =', os.environ.get('TORCH_CUDA_ARCH_LIST'))
9
  os.system("python -m pip install git+https://github.com/YixunLiang/simple-knn.git")
10
+ print('==> simple-knn installed!')
11
  os.system("python -m pip install git+https://github.com/YixunLiang/diff-gaussian-rasterization.git")
12
+ print('==> diff-gaussian-rasterization installed!')
13
+ else:
14
+ print('==> simple-knn & diff-gaussian-rasterization already installed!')
15
 
16
  from train import *
17