在运行main_test_face_enhancement.py的时候会出现编译自定义 CUDA 代码 (fused_bias_act_kernel.cu) 时失败

[yyysiyue1]

README

MD

E:\KAIR-master>python main_test_face_enhancement.py

D:\Python3\Lib\site-packages\torch\utils\cpp_extension.py:2356: UserWarning: TORCH_CUDA_ARCH_LIST is not set, all archs for visible cards are included for compilation.

If this is not desired, please set os.environ['TORCH_CUDA_ARCH_LIST'].

  warnings.warn(

Traceback (most recent call last):

  File "D:\Python3\Lib\site-packages\torch\utils\cpp_extension.py", line 2506, in _run_ninja_build

    subprocess.run(

    ~~~~~~~~~~~~~~^

        command,

        ^^^^^^^^

    ...<3 lines>...

        check=True,

        ^^^^^^^^^^^

        env=env)

        ^^^^^^^^

  File "D:\Python3\Lib\subprocess.py", line 577, in run

    raise CalledProcessError(retcode, process.args,

                             output=stdout, stderr=stderr)

subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):

  File "E:\KAIR-master\main_test_face_enhancement.py", line 45, in

    from models.network_faceenhancer import FullGenerator as enhancer_net

  File "E:\KAIR-master\models\network_faceenhancer.py", line 10, in

    from op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d

  File "E:\KAIR-master\models\op_init_.py", line 1, in

    from .fused_act import FusedLeakyReLU, fused_leaky_relu

  File "E:\KAIR-master\models\op\fused_act.py", line 10, in

    fused = load(

        'fused',

    ...<3 lines>...

        ],

    )

  File "D:\Python3\Lib\site-packages\torch\utils\cpp_extension.py", line 1623, in load

    return _jit_compile(

        name,

    ...<11 lines>...

        is_standalone,

        keep_intermediates=keep_intermediates)

  File "D:\Python3\Lib\site-packages\torch\utils\cpp_extension.py", line 2076, in _jit_compile

    _write_ninja_file_and_build_library(

    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^

        name=name,

        ^^^^^^^^^^

    ...<9 lines>...

        with_sycl=with_sycl,

        ^^^^^^^^^^^^^^^^^^^^

        is_standalone=is_standalone)

        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^

  File "D:\Python3\Lib\site-packages\torch\utils\cpp_extension.py", line 2222, in _write_ninja_file_and_build_library

    _run_ninja_build(

    ~~~~~~~~~~~~~~~~^

        build_directory,

        ^^^^^^^^^^^^^^^^

        verbose,

        ^^^^^^^^

        error_prefix=f"Error building extension '{name}'")

        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

  File "D:\Python3\Lib\site-packages\torch\utils\cpp_extension.py", line 2522, in _run_ninja_build

    raise RuntimeError(message) from e

RuntimeError: Error building extension 'fused': [1/3] C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin\nvcc --generate-dependencies-with-compile --dependency-output fused_bias_act_kernel.cuda.o.d -Xcudafe --diag_suppress=dll_interface_conflict_dllexport_assumed -Xcudafe --diag_suppress=dll_interface_conflict_none_assumed -Xcudafe --diag_suppress=field_without_dll_interface -Xcudafe --diag_suppress=base_class_has_different_dll_interface -Xcompiler /EHsc -Xcompiler /wd4068 -Xcompiler /wd4067 -Xcompiler /wd4624 -Xcompiler /wd4190 -Xcompiler /wd4018 -Xcompiler /wd4275 -Xcompiler /wd4267 -Xcompiler /wd4244 -Xcompiler /wd4251 -Xcompiler /wd4819 -Xcompiler /MD -DTORCH_EXTENSION_NAME=fused -DTORCH_API_INCLUDE_EXTENSION_H -ID:\Python3\Lib\site-packages\torch\include -ID:\Python3\Lib\site-packages\torch\include\torch\csrc\api\include "-IC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include" -ID:\Python3\Include -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=compute_61 -gencode=arch=compute_61,code=sm_61 -std=c++17 -c E:\KAIR-master\models\op\fused_bias_act_kernel.cu -o fused_bias_act_kernel.cuda.o

FAILED: fused_bias_act_kernel.cuda.o