No module named 'upfirdn2d_cuda'

NVlabs / imaginaire

NVIDIA's Deep Imagination Team's PyTorch Library

Other

4k stars 448 forks source link

Closed yangwenbing2 closed 2 years ago

yangwenbing2 commented 2 years ago

when i run python -m torch.distributed.launch --nproc_per_node=8 train.py --config configs/projects/munit/afhq_dog2cat/ampO1.yaml

the error raise.

my enviroment: cuda: 10.2 pytorch:1.7 python:3.6

mingyuliutw commented 2 years ago

upfirdn2d_cuda will be compiled when you run scripts/install.sh