cd selective_scan &&pip install . 一直报错

[LukasYan30]

报错信息如下：Processing /data/yhl/mamba/Mamba-YOLO-main/selective_scan
Preparing metadata (setup.py) ... done
Requirement already satisfied: torch in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from selective_scan==0.0.2) (2.3.1+cu118)
Collecting packaging (from selective_scan==0.0.2)
Using cached packaging-24.1-py3-none-any.whl.metadata (3.2 kB)
Collecting ninja (from selective_scan==0.0.2)
Using cached ninja-1.11.1.1-py2.py3-none-manylinux1_x86_64.manylinux_2_5_x86_64.whl.metadata (5.3 kB)
Collecting einops (from selective_scan==0.0.2)
Using cached einops-0.8.0-py3-none-any.whl.metadata (12 kB)
Requirement already satisfied: filelock in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (3.15.4)
Requirement already satisfied: typing-extensions>=4.8.0 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (4.12.2)
Requirement already satisfied: sympy in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (1.13.1)
Requirement already satisfied: networkx in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (3.3)
Requirement already satisfied: jinja2 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (3.1.4)
Requirement already satisfied: fsspec in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (2024.6.1)
Requirement already satisfied: nvidia-cuda-nvrtc-cu11==11.8.89 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (11.8.89)
Requirement already satisfied: nvidia-cuda-runtime-cu11==11.8.89 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (11.8.89)
Requirement already satisfied: nvidia-cuda-cupti-cu11==11.8.87 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (11.8.87)
Requirement already satisfied: nvidia-cudnn-cu11==8.7.0.84 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (8.7.0.84)
Requirement already satisfied: nvidia-cublas-cu11==11.11.3.6 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (11.11.3.6)
Requirement already satisfied: nvidia-cufft-cu11==10.9.0.58 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (10.9.0.58)
Requirement already satisfied: nvidia-curand-cu11==10.3.0.86 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (10.3.0.86)
Requirement already satisfied: nvidia-cusolver-cu11==11.4.1.48 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (11.4.1.48)
Requirement already satisfied: nvidia-cusparse-cu11==11.7.5.86 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (11.7.5.86)
Requirement already satisfied: nvidia-nccl-cu11==2.20.5 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (2.20.5)
Requirement already satisfied: nvidia-nvtx-cu11==11.8.86 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (11.8.86)
Requirement already satisfied: triton==2.3.1 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from torch->selective_scan==0.0.2) (2.3.1)
Requirement already satisfied: MarkupSafe>=2.0 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from jinja2->torch->selective_scan==0.0.2) (2.1.5)
Requirement already satisfied: mpmath<1.4,>=1.1.0 in /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages (from sympy->torch->selective_scan==0.0.2) (1.3.0)
Using cached einops-0.8.0-py3-none-any.whl (43 kB)
Using cached ninja-1.11.1.1-py2.py3-none-manylinux1_x86_64.manylinux_2_5_x86_64.whl (307 kB)
Using cached packaging-24.1-py3-none-any.whl (53 kB)
Building wheels for collected packages: selective_scan
Building wheel for selective_scan (setup.py) ... error
error: subprocess-exited-with-error

× python setup.py bdist_wheel did not run successfully.
│ exit code: 1
╰─> [32 lines of output]

  torch.__version__  = 2.3.1+cu118
  
  
  
  
  CUDA_HOME = /root/miniconda3/envs/mambayolo
  
  
  CUDA version:  11.8
  running bdist_wheel
  /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages/torch/utils/cpp_extension.py:499: 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.'))
  running build
  running build_ext
  /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages/torch/utils/cpp_extension.py:428: UserWarning: There are no g++ version bounds defined for CUDA version 11.8
    warnings.warn(f'There are no {compiler_name} version bounds defined for CUDA version {cuda_str_version}')
  building 'selective_scan_cuda_core' extension
  creating build
  creating build/temp.linux-x86_64-cpython-310
  creating build/temp.linux-x86_64-cpython-310/csrc
  creating build/temp.linux-x86_64-cpython-310/csrc/selective_scan
  creating build/temp.linux-x86_64-cpython-310/csrc/selective_scan/cus
  gcc -pthread -B /root/miniconda3/envs/mambayolo/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -fwrapv -O2 -Wall -fPIC -O2 -isystem /root/miniconda3/envs/mambayolo/include -fPIC -O2 -isystem /root/miniconda3/envs/mambayolo/include -fPIC -I/data/yhl/mamba/Mamba-YOLO-main/selective_scan/csrc/selective_scan -I/root/miniconda3/envs/mambayolo/lib/python3.10/site-packages/torch/include -I/root/miniconda3/envs/mambayolo/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/root/miniconda3/envs/mambayolo/lib/python3.10/site-packages/torch/include/TH -I/root/miniconda3/envs/mambayolo/lib/python3.10/site-packages/torch/include/THC -I/root/miniconda3/envs/mambayolo/include -I/root/miniconda3/envs/mambayolo/include/python3.10 -c csrc/selective_scan/cus/selective_scan.cpp -o build/temp.linux-x86_64-cpython-310/csrc/selective_scan/cus/selective_scan.o -O3 -std=c++17 -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -DTORCH_EXTENSION_NAME=selective_scan_cuda_core -D_GLIBCXX_USE_CXX11_ABI=0
  In file included from /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages/torch/include/ATen/cuda/CUDAContext.h:3,
                   from csrc/selective_scan/cus/selective_scan.cpp:5:
  /root/miniconda3/envs/mambayolo/lib/python3.10/site-packages/torch/include/ATen/cuda/CUDAContextLight.h:6:10: fatal error: cuda_runtime_api.h: No such file or directory
      6 | #include <cuda_runtime_api.h>
        |          ^~~~~~~~~~~~~~~~~~~~
  compilation terminated.
  error: command '/root/miniconda3/envs/mambayolo/bin/gcc' failed with exit code 1
  [end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed building wheel for selective_scan
Running setup.py clean for selective_scan
Failed to build selective_scan
ERROR: Could not build wheels for selective_scan, which is required to install pyproject.toml-based projects 我已正确安装了gcc11版本的，但是还是未解决这个问题。

[Shen-LW]

selective_scan来自vmanba，解决可以参考：https://blog.csdn.net/yyywxk/article/details/140422758

[DawnMagnet]

#65