Spaces:
Runtime error
Runtime error
Update gradio_demo.py
Browse files- gradio_demo.py +1 -1
gradio_demo.py
CHANGED
@@ -4,7 +4,7 @@ import os
|
|
4 |
from train import *
|
5 |
|
6 |
# https://github.com/pytorch/extension-cpp/issues/71
|
7 |
-
if
|
8 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX"
|
9 |
os.system("python -m pip install git+https://github.com/YixunLiang/simple-knn.git")
|
10 |
os.system("python -m pip install git+https://github.com/YixunLiang/diff-gaussian-rasterization.git")
|
|
|
4 |
from train import *
|
5 |
|
6 |
# https://github.com/pytorch/extension-cpp/issues/71
|
7 |
+
if os.environ.get('TORCH_CUDA_ARCH_LIST') != "3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX":
|
8 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX"
|
9 |
os.system("python -m pip install git+https://github.com/YixunLiang/simple-knn.git")
|
10 |
os.system("python -m pip install git+https://github.com/YixunLiang/diff-gaussian-rasterization.git")
|