Spaces:
Runtime error
Runtime error
Update gradio_demo.py
Browse files- 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 |
|