haodongli commited on
Commit
666f805
·
verified ·
1 Parent(s): dc82dbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ 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"] = "8.6+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!')
 
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!')