fatal error: cuda.h: 没有那个文件或目录（ pcdet/ops/iou3d_nms/src/iou3d_cpu.cpp:12:18: fatal error: cuda.h: ）

最新推荐文章于 2025-05-07 11:54:24 发布

hello689

最新推荐文章于 2025-05-07 11:54:24 发布

阅读量7.0k

点赞数 11

CC 4.0 BY-SA版权

分类专栏： 3D目标检测文章标签： CaDDN 单目3D目标检测 pcdet cuda

本文链接：https://blog.csdn.net/jiachang98/article/details/121933403

3D目标检测专栏收录该内容

7 篇文章

订阅专栏

在编译PCDet时遇到找不到cuda.h的错误，尝试了修改环境变量和添加include_dirs两种方法。首先检查环境变量设置，如果nvcc正常，可能是缺少CUDA头文件。通过在setup.py中添加include_dirs解决，将CUDA的include路径包含进去。若问题依旧，可能需要进一步检查代码或环境配置。提供的博客链接提供了更多解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、问题描述

我是编译pcdet的时候，执行命令python setup.py develop，出现了如下这样的问题：

gcc -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -fPIC -I/opt/python3.7/lib/python3.7/site-packages/torch/include -I/opt/python3.7/lib/python3.7/site-packages/torch/include/torch/csrc/api/include -I/opt/python3.7/lib/python3.7/site-packages/torch/include/TH -I/opt/python3.7/lib/python3.7/site-packages/torch/include/THC -I/usr/local/cuda/:/usr/local/cuda-10.1/:/usr/local/cuda-10.1:/usr/local/cuda-10.1::/usr/local/cuda-10.1/include -I/opt/python3.7/include/python3.7m -c pcdet/ops/iou3d_nms/src/iou3d_cpu.cpp -o build/temp.linux-x86_64-3.7/pcdet/ops/iou3d_nms/src/iou3d_cpu.o -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=iou3d_nms_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++11 pcdet/ops/iou3d_nms/src/iou3d_cpu.cpp:12:18: fatal error: cuda.h: 没有这样的文件或目录 compilation terminated. error: command ‘gcc’ failed with exit status 1

注当时没有截图，后来怎么改都不会再出现这个问题了，但我记得不错的话，就是类似上边的这种错误。

2、问题解决

很多博客给出的都是环境变量设置的解决方法，可我的环境变量明明对了呀，nvcc -V也是正常运行的。但就是一直出现找不到cuda.h的问题，我也曾把/usr/local/cuda/inclde中的cuda.h拷贝过去，发现不现实，需要拷贝的东西很多，放弃了。
不过你也可以试试修改一下环境变量，也有可能还就是环境变量的问题，不过我建议先执行b步骤，b不行的话再来执行a步骤。

a.修改环境变量

这里我推荐几个参考博客吧，不在赘述了。

b.添加include_dirs

这里参考了这篇博客
在setup.py文件中做出如下修改即可：

添加代码

include_dirs = [os.path.realpath('../include'), '/usr/local/cuda/include/']

修改make_cuda_ext函数：

def make_cuda_ext(name, module, sources):
    cuda_ext = CUDAExtension(
        name='%s.%s' % (module, name),
        sources=[os.path.join(*module.split('.'), src) for src in sources],
        include_dirs=include_dirs
    )
    return cuda_ext

如下图所示：
在这里插入图片描述如果出现了:/usr/local/cuda/bin/nvcc的问题，那确实需要修改环境变量，可参考这个博客

3、还没能解决问题的话

是否是代码需要修改，这里可以参考W3cschool中关于pytorch的PyTorch torch.utils.cpp_extension的使用，希望能够帮到你，有时候静下心来看一看错误，看一看源码问题可能就解决了。
https://www.w3cschool.cn/pytorch/pytorch-dhfk3bxl.html

致谢博客：
https://blog.csdn.net/qq_39031960/article/details/106211695
https://blog.csdn.net/dongwanli666/article/details/104656679