Update app.py

app.py

@@ -3,7 +3,7 @@ import numpy as np
 import os
 
 import torch
-print(torch.cuda.get_device_capability(0))
+print("CUDA_ARCH ==>", torch.cuda.get_device_capability(0))
 
 try:
     from train import *
@@ -11,7 +11,7 @@ try:
 except:
     print('==> simple-knn & diff-gaussian-rasterization are NOT installed!')
     # https://github.com/pytorch/extension-cpp/issues/71
-    os.environ["TORCH_CUDA_ARCH_LIST"] = "3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX"
+    os.environ["TORCH_CUDA_ARCH_LIST"] = "8.6"
     print('==> TORCH_CUDA_ARCH_LIST =', os.environ.get('TORCH_CUDA_ARCH_LIST'))
     os.system("python -m pip install git+https://github.com/openai/point-e.git")
     print('==> point-e installed!')