编译程序报错IndexError: list index out of range

在github下载mish-cuda，下载地址：https://github.com/JunnYu/mish-cuda
进行编译，但是在编译过程中报错
编译命令：sudo python3.7 setup.py install


报错信息：

nvidia@miivii-tegra:/opts/mish-cuda$ sudo python3.7 setup.py build

No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda'

/usr/lib/python3.7/distutils/extension.py:131: UserWarning: Unknown Extension options: 'headers'

  warnings.warn(msg)

running build

running build_py

running egg_info

writing src/mish_cuda.egg-info/PKG-INFO

writing dependency_links to src/mish_cuda.egg-info/dependency_links.txt

writing requirements to src/mish_cuda.egg-info/requires.txt

writing top-level names to src/mish_cuda.egg-info/top_level.txt

/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py:369: UserWarning: Attempted to use ninja as the BuildExtension backend but we could not find ninja.. Falling back to using the slow distutils backend.

  warnings.warn(msg.format('we could not find ninja.'))

reading manifest file 'src/mish_cuda.egg-info/SOURCES.txt'

writing manifest file 'src/mish_cuda.egg-info/SOURCES.txt'

running build_ext

building 'mish_cuda._C' extension

aarch64-linux-gnu-gcc -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/usr/local/lib/python3.7/dist-packages/torch/include -I/usr/local/lib/python3.7/dist-packages/torch/include/torch/csrc/api/include -I/usr/local/lib/python3.7/dist-packages/torch/include/TH -I/usr/local/lib/python3.7/dist-packages/torch/include/THC -I/usr/local/cuda/include -I/usr/include/python3.7m -c csrc/cpu/mish_cpu.cpp -o build/temp.linux-aarch64-3.7/csrc/cpu/mish_cpu.o -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="_gcc" -DPYBIND11_STDLIB="_libstdcpp" -DPYBIND11_BUILD_ABI="_cxxabi1013" -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14

In file included from /usr/local/lib/python3.7/dist-packages/torch/include/ATen/Parallel.h:140:0,

                 from /usr/local/lib/python3.7/dist-packages/torch/include/ATen/CPUApplyUtils.h:3,

                 from csrc/cpu/mish_cpu.cpp:3:

/usr/local/lib/python3.7/dist-packages/torch/include/ATen/ParallelOpenMP.h:83:0: warning: ignoring #pragma omp parallel [-Wunknown-pragmas]

#pragma omp parallel for if ((end - begin) >= grain_size)

aarch64-linux-gnu-gcc -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/usr/local/lib/python3.7/dist-packages/torch/include -I/usr/local/lib/python3.7/dist-packages/torch/include/torch/csrc/api/include -I/usr/local/lib/python3.7/dist-packages/torch/include/TH -I/usr/local/lib/python3.7/dist-packages/torch/include/THC -I/usr/local/cuda/include -I/usr/include/python3.7m -c csrc/cuda/mish_cuda.cpp -o build/temp.linux-aarch64-3.7/csrc/cuda/mish_cuda.o -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="_gcc" -DPYBIND11_STDLIB="_libstdcpp" -DPYBIND11_BUILD_ABI="_cxxabi1013" -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14

In file included from /usr/local/lib/python3.7/dist-packages/torch/include/ATen/Parallel.h:140:0,

                 from /usr/local/lib/python3.7/dist-packages/torch/include/torch/csrc/api/include/torch/utils.h:3,

                 from /usr/local/lib/python3.7/dist-packages/torch/include/torch/csrc/api/include/torch/nn/cloneable.h:5,

                 from /usr/local/lib/python3.7/dist-packages/torch/include/torch/csrc/api/include/torch/nn.h:3,

                 from /usr/local/lib/python3.7/dist-packages/torch/include/torch/csrc/api/include/torch/all.h:13,

                 from /usr/local/lib/python3.7/dist-packages/torch/include/torch/extension.h:4,

                 from csrc/cuda/mish_cuda.cpp:2:

/usr/local/lib/python3.7/dist-packages/torch/include/ATen/ParallelOpenMP.h:83:0: warning: ignoring #pragma omp parallel [-Wunknown-pragmas]

#pragma omp parallel for if ((end - begin) >= grain_size)

Traceback (most recent call last):

  File "setup.py", line 29, in <module>

    cmdclass={'build_ext': BuildExtension})

  File "/usr/lib/python3/dist-packages/setuptools/__init__.py", line 129, in setup

    return distutils.core.setup(**attrs)

  File "/usr/lib/python3.7/distutils/core.py", line 148, in setup

    dist.run_commands()

  File "/usr/lib/python3.7/distutils/dist.py", line 966, in run_commands

    self.run_command(cmd)

  File "/usr/lib/python3.7/distutils/dist.py", line 985, in run_command

    cmd_obj.run()

  File "/usr/lib/python3.7/distutils/command/build.py", line 135, in run

    self.run_command(cmd_name)

  File "/usr/lib/python3.7/distutils/cmd.py", line 313, in run_command

    self.distribution.run_command(command)

  File "/usr/lib/python3.7/distutils/dist.py", line 985, in run_command

    cmd_obj.run()

  File "/usr/lib/python3/dist-packages/setuptools/command/build_ext.py", line 78, in run

    _build_ext.run(self)

  File "/usr/lib/python3.7/distutils/command/build_ext.py", line 340, in run

    self.build_extensions()

  File "/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py", line 708, in build_extensions

    build_ext.build_extensions(self)

  File "/usr/lib/python3.7/distutils/command/build_ext.py", line 449, in build_extensions

    self._build_extensions_serial()

  File "/usr/lib/python3.7/distutils/command/build_ext.py", line 474, in _build_extensions_serial

    self.build_extension(ext)

  File "/usr/lib/python3/dist-packages/setuptools/command/build_ext.py", line 199, in build_extension

    _build_ext.build_extension(self, ext)

  File "/usr/lib/python3.7/distutils/command/build_ext.py", line 534, in build_extension

    depends=ext.depends)

  File "/usr/lib/python3.7/distutils/ccompiler.py", line 574, in compile

    self._compile(obj, src, ext, cc_args, extra_postargs, pp_opts)

  File "/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py", line 456, in unix_wrap_single_compile

    cflags = unix_cuda_flags(cflags)

  File "/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py", line 423, in unix_cuda_flags

    cflags + _get_cuda_arch_flags(cflags))

  File "/usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py", line 1561, in _get_cuda_arch_flags

    arch_list[-1] += '+PTX'

IndexError: list index out of range

1780675810 发表于 2021-3-16 17:52
Traceback (most recent call last):

  File "setup.py", line 29, in

你好，对于github项目的编译，应该是开源代码本身存在问题。
建议您可以在github项目的，issue部分提问。
或者可以选择其他的开源项目。

这个程序在x86上就能编译通过，在arm上就报错

arm机器上cuda是10.2  torch是1.8.0  tourch和cuda版本不匹配