search

qinglew / PCN-PyTorch

Implementation of PCN(Point Completion Network) in PyTorch.
155 stars 33 forks source link

Colab run PCN-Pytorch #14

Open kafei123456 opened 2 years ago

kafei123456 commented 2 years ago

colab:CUDA-10.1 ,pytorch1.7, python3.7 the trackback: Traceback (most recent call last): File "train.py", line 14, in from metrics.metric import l1_cd File "/content/drive/MyDrive/PCN-PyTorch/metrics/metric.py", line 5, in from extensions.earth_movers_distance.emd import EarthMoverDistance File "/content/drive/MyDrive/PCN-PyTorch/extensions/earth_movers_distance/emd.py", line 3, in import emd_cuda ModuleNotFoundError: No module named 'emd_cuda'

I don't know the problom, when run PCN-pytorch in colab.and I have compiled the "earth_movers_distance" core.

Are there any solutions?

hiepbk commented 2 years ago

I think you haven't complied for emd and cd, refer this for colab: %cd extensions/chamfer_distance !python setup.py install %cd ../earth_movers_distance !python setup.py install

kafei123456 commented 2 years ago

Think you for your valuable advices.