Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
15 |
-
|
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")
|