Spaces:
Runtime error
Runtime error
torch-cluster was without cuda support
Browse files- app.py +1 -1
- requirements.txt +1 -2
app.py
CHANGED
@@ -24,7 +24,7 @@ def run_on_gpu(input_point_cloud: gr.utils.NamedString,
|
|
24 |
import os
|
25 |
import sys
|
26 |
sys.path.append(os.path.abspath('ppsurf'))
|
27 |
-
from ppsurf.pps import cli_main
|
28 |
import subprocess
|
29 |
|
30 |
in_file = '{}'.format(input_point_cloud.name)
|
|
|
24 |
import os
|
25 |
import sys
|
26 |
sys.path.append(os.path.abspath('ppsurf'))
|
27 |
+
# from ppsurf.pps import cli_main
|
28 |
import subprocess
|
29 |
|
30 |
in_file = '{}'.format(input_point_cloud.name)
|
requirements.txt
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
pytorch-lightning>=2.1
|
2 |
torch-geometric>=2
|
3 |
-
|
4 |
-
torch-cluster>=1.6.3
|
5 |
scikit-learn-intelex>=2024.1.0
|
6 |
numpy>=1.26.4
|
7 |
scikit-image>=0.22.0
|
|
|
1 |
pytorch-lightning>=2.1
|
2 |
torch-geometric>=2
|
3 |
+
torch-cluster>=1.6.3 -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
|
|
|
4 |
scikit-learn-intelex>=2024.1.0
|
5 |
numpy>=1.26.4
|
6 |
scikit-image>=0.22.0
|