haodongli commited on
Commit
29c5542
·
1 Parent(s): 60f932e

Update gradio_demo.py

Browse files
Files changed (1) hide show
  1. 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 value = 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")
 
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")