search

uber-research / GTN

Other
76 stars 28 forks source link

ModuleNotFoundError: No module named 'custom_backward_cpp' #1

Open jayurbain opened 4 years ago

jayurbain commented 4 years ago

Environment: MacOS 10.13.6 Installed pytorch==1.2, mpi4py, fire with conda. horovod with pip

Error: $ python train_cgtn.py . Traceback (most recent call last): File "train_cgtn.py", line 24, in from models import Classifier, Generator, Encoder, sample_model File "/Users/jayurbain/Dropbox/GTN/models.py", line 479, in from exconv import ExConv2d File "/Users/jayurbain/Dropbox/GTN/exconv.py", line 16, in import custom_backward_cpp ModuleNotFoundError: No module named 'custom_backward_cpp'

Any direction would be appreciated. Thanks, Jay

public-daniel commented 4 years ago

You just need to run

python setup.py install

Checkout this guide from PyTorch for some additional details: https://brsoff.github.io/tutorials/advanced/cpp_extension.html