Open wwt123-star opened 1 year ago
i hava the same question
I got this problem too
warnings.warn('Failed to build CUDA kernels for upfirdn2d. Falling back to slow reference implementation. Details:\n\n' + traceback.format_exc()) Setting up PyTorch plugin "upfirdn2d_plugin"... Failed! /content/drive/MyDrive/StyleGAN/stylegan2-ada-pytorch/torch_utils/ops/upfirdn2d.py:34: UserWarning: Failed to build CUDA kernels for upfirdn2d. Falling back to slow reference implementation. Details:
Traceback (most recent call last): File "/content/drive/MyDrive/StyleGAN/stylegan2-ada-pytorch/torch_utils/ops/upfirdn2d.py", line 32, in _init _plugin = custom_ops.get_plugin('upfirdn2d_plugin', sources=sources, extra_cuda_cflags=['--use_fast_math']) File "/content/drive/MyDrive/StyleGAN/stylegan2-ada-pytorch/torch_utils/custom_ops.py", line 110, in get_plugin torch.utils.cpp_extension.load(name=module_name, verbose=verbose_build, sources=sources, **build_kwargs) File "/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py", line 997, in load keep_intermediates=keep_intermediates) File "/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py", line 1213, in _jit_compile return _import_module_from_library(name, build_directory, is_python_module) File "/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py", line 1560, in _import_module_from_library file, path, description = imp.find_module(module_name, [path]) File "/usr/lib/python3.7/imp.py", line 296, in find_module raise ImportError(_ERR_MSG.format(name), name=name) ImportError: No module named 'upfirdn2d_plugin'
I got this problem too
warnings.warn('Failed to build CUDA kernels for upfirdn2d. Falling back to slow reference implementation. Details:\n\n' + traceback.format_exc()) Setting up PyTorch plugin "upfirdn2d_plugin"... Failed! /content/drive/MyDrive/StyleGAN/stylegan2-ada-pytorch/torch_utils/ops/upfirdn2d.py:34: UserWarning: Failed to build CUDA kernels for upfirdn2d. Falling back to slow reference implementation. Details:
Traceback (most recent call last): File "/content/drive/MyDrive/StyleGAN/stylegan2-ada-pytorch/torch_utils/ops/upfirdn2d.py", line 32, in _init _plugin = custom_ops.get_plugin('upfirdn2d_plugin', sources=sources, extra_cuda_cflags=['--use_fast_math']) File "/content/drive/MyDrive/StyleGAN/stylegan2-ada-pytorch/torch_utils/custom_ops.py", line 110, in get_plugin torch.utils.cpp_extension.load(name=module_name, verbose=verbose_build, sources=sources, **build_kwargs) File "/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py", line 997, in load keep_intermediates=keep_intermediates) File "/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py", line 1213, in _jit_compile return _import_module_from_library(name, build_directory, is_python_module) File "/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py", line 1560, in _import_module_from_library file, path, description = imp.find_module(module_name, [path]) File "/usr/lib/python3.7/imp.py", line 296, in find_module raise ImportError(_ERR_MSG.format(name), name=name) ImportError: No module named 'upfirdn2d_plugin'
have you solved it yet
When I run the train.py, it reports the ImportError of *_plugin. I had found the 'build.ninja' generated in the 'bias_act_plugin' and 'upfirdn2d_plugin' files but no 'bias_act.o', 'bias_act.cuda.o' and 'bias_act_plugin.pyd'. How can I solve these issues?