Mamba模型使用报错—— fatal error: cuda_runtime_api.h: No such file or directory

训练Mamba模型的时候执行环境安装命令的时候报错这个问题

fatal error: cuda_runtime_api.h: No such file or directory
    6 | #include <cuda_runtime_api.h>
      |          ^~~~~~~~~~~~~~~~~~~~
compilation terminated.

报错是说找不到cuda的一个运行接口头文件，且出现在gcc编译过程中，于是重点关注这一块内容。

首先要有安装了cuda，否者是不会有这个cuda_runtime_api.h这个文件滴，如果安装了之后还是报这个错，那么就在用户目录下面的.bashrc文件最后加上以下片段：

export CUDA=xx.x
export PATH=/usr/local/cuda-$CUDA/bin:$PATH
export CUDA_PATH=/usr/local/cuda-$CUDA
export CUDA_HOME=/usr/local/cuda-$CUDA
export LIBRARY_PATH=$CUDA_HOME/lib64:$LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda-$CUDA/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH

这里如果安装了cuda的话，就会有/usr/local/cuda-xxxxx的目录，选择对应的版本即可，完美解决~

参考自：Fatal error: cuda_runtime_api.h not found - Jetson & Embedded Systems / Jetson AGX Xavier - NVIDIA Developer Forums