perler commited on
Commit
8d71fde
1 Parent(s): 028cb6a

torch-cluster was without cuda support

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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
- #torch-cluster -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
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