No module named 'upfirdn2d_cuda'

[IUUI11]

when I run this command "python -m torch.distributed.launch --nproc_per_node=8 train.py
--config configs/projects/munit/afhq_dog2cat/ampO1.yaml" or "python -m torch.distributed.launch --nproc_per_node=1 inference.py --config configs/projects/munit/afhq_dog2cat/ampO1.yaml --output_dir projects/munit/output/afhq_dog2cat" , always said

[IUUI11]

[MeimShang]

You should install Imaginaire at first, follow the install instruction in readme.md

[Gienapp]

For me the solution was to activate the conda environment before installing the third party libraries.

[ingaj89]

Your modules may be build to the “wrong” location so python doesn’t find them. You can add “—user” to the python setup.py install command so the modules will be installed to /use/lib/python/dist-packages.
Or look at other options to set a custom install path for setuptools.