haodongli commited on
Commit
43fa63a
·
verified ·
1 Parent(s): c2a54e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ try:
11
  except:
12
  print('==> simple-knn & diff-gaussian-rasterization are NOT installed!')
13
  # https://github.com/pytorch/extension-cpp/issues/71
14
- # os.environ["TORCH_CUDA_ARCH_LIST"] = "5.0;5.2;5.3;6.0;6.1;6.2;7.0;7.2;7.5;8.0;8.6;8.7;8.9;9.0+PTX"
15
- # print('==> TORCH_CUDA_ARCH_LIST =', os.environ.get('TORCH_CUDA_ARCH_LIST'))
16
  os.system("python -m pip install git+https://github.com/openai/point-e.git")
17
  print('==> point-e installed!')
18
  os.system("python -m pip install git+https://github.com/YixunLiang/simple-knn.git")
 
11
  except:
12
  print('==> simple-knn & diff-gaussian-rasterization are NOT installed!')
13
  # https://github.com/pytorch/extension-cpp/issues/71
14
+ os.environ["TORCH_CUDA_ARCH_LIST"] = "5.0;5.2;5.3;6.0;6.1;6.2;7.0;7.2;7.5;8.0;8.6;8.7;8.9;9.0+PTX"
15
+ print('==> TORCH_CUDA_ARCH_LIST =', os.environ.get('TORCH_CUDA_ARCH_LIST'))
16
  os.system("python -m pip install git+https://github.com/openai/point-e.git")
17
  print('==> point-e installed!')
18
  os.system("python -m pip install git+https://github.com/YixunLiang/simple-knn.git")