解决:src/cpp/cuda.hpp:fatal error: cuda.h

本文档介绍了如何解决CUDA编程时遇到的致命错误:`src/cpp/cuda.hpp:fatal error: cuda.h`。关键在于正确配置`.bashrc`文件,确保系统路径包含CUDA库的头文件路径。

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

  • 在.bashrc中,如下修改:
vi ~/.bashrc
export PATH=/usr/local/cuda-10.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
  • 生效
bash ~/.bashrc
#source ~/.bashrc

 

blackyin@yin:~/Desktop/tensorRT_Pro-YOLOv8-main$ make rtdetr -j64 Compile depends CUDA src/tensorRT/onnxplugin/plugins/HSwish.cu Compile depends CUDA src/tensorRT/onnxplugin/plugins/ScatterND.cu Compile depends CUDA src/tensorRT/onnxplugin/plugins/HSigmoid.cu Compile depends CUDA src/tensorRT/onnxplugin/plugins/DCNv2.cu Compile depends CXX src/tensorRT/builder/trt_builder.cpp Compile depends CXX src/tensorRT/onnxplugin/onnxplugin.cpp Compile depends CXX src/tensorRT/infer/trt_infer.cpp Compile depends CXX src/tensorRT/onnx_parser/ShapedWeights.cpp Compile depends CXX src/tensorRT/onnx_parser/OnnxAttrs.cpp Compile depends CXX src/tensorRT/onnx_parser/LoopHelpers.cpp Compile depends CXX src/tensorRT/onnx_parser/builtin_op_importers.cpp Compile depends CXX src/tensorRT/onnx_parser/onnxErrorRecorder.cpp Compile depends CXX src/tensorRT/onnx_parser/RNNHelpers.cpp Compile depends CXX src/tensorRT/onnx_parser/ModelImporter.cpp Compile depends CXX src/tensorRT/onnx_parser/ShapeTensor.cpp Compile depends CXX src/tensorRT/onnx_parser/NvOnnxParser.cpp Compile depends CXX src/tensorRT/onnx_parser/onnx2trt_utils.cpp In file included from src/tensorRT/onnx_parser/OnnxAttrs.cpp:5: src/tensorRT/onnx_parser/OnnxAttrs.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/LoopHelpers.cpp:5: src/tensorRT/onnx_parser/LoopHelpers.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/ShapedWeights.cpp:5: src/tensorRT/onnx_parser/ShapedWeights.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/onnx2trt.hpp:7, from src/tensorRT/onnx_parser/builtin_op_importers.hpp:7, from src/tensorRT/onnx_parser/builtin_op_importers.cpp:5: src/tensorRT/onnx_parser/NvOnnxParser.h:8:10: fatal error: NvInfer.h: No such file or directory 8 | #include "NvInfer.h" | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/onnxErrorRecorder.cpp:5: src/tensorRT/onnx_parser/onnxErrorRecorder.hpp:7:10: fatal error: NvInferRuntimeCommon.h: No such file or directory 7 | #include "NvInferRuntimeCommon.h" | ^~~~~~~~~~~~~~~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/RNNHelpers.cpp:5: src/tensorRT/onnx_parser/RNNHelpers.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/onnx2trt.hpp:7, from src/tensorRT/onnx_parser/ImporterContext.hpp:7, from src/tensorRT/onnx_parser/ModelImporter.hpp:7, from src/tensorRT/onnx_parser/ModelImporter.cpp:5: src/tensorRT/onnx_parser/NvOnnxParser.h:8:10: fatal error: NvInfer.h: No such file or directory 8 | #include "NvInfer.h" | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/ShapeTensor.cpp:5: src/tensorRT/onnx_parser/ShapeTensor.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/NvOnnxParser.cpp:5: src/tensorRT/onnx_parser/NvOnnxParser.h:8:10: fatal error: NvInfer.h: No such file or directory 8 | #include "NvInfer.h" | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/onnx2trt_utils.hpp:7, from src/tensorRT/onnx_parser/onnx2trt_utils.cpp:5: src/tensorRT/onnx_parser/ShapedWeights.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnxplugin/onnxplugin.cpp:2: src/tensorRT/onnxplugin/onnxplugin.hpp:9:10: fatal error: NvInfer.h: No such file or directory 9 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. src/tensorRT/infer/trt_infer.cpp:6:10: fatal error: NvInfer.h: No such file or directory 6 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnxplugin/plugins/ScatterND.cu:2: src/tensorRT/onnxplugin/onnxplugin.hpp:9:10: fatal error: NvInfer.h: No such file or directory 9 | #include <NvInfer.h> | ^~~~~~~~~~~ In file included from src/tensorRT/onnxplugin/plugins/DCNv2.cu:3: src/tensorRT/onnxplugin/onnxplugin.hpp:9:10: fatal error: NvInfer.h: No such file or directory 9 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. compilation terminated. In file included from src/tensorRT/onnxplugin/plugins/HSwish.cu:2: src/tensorRT/onnxplugin/onnxplugin.hpp:9:10: fatal error: NvInfer.h: No such file or directory 9 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. src/tensorRT/builder/trt_builder.cpp:6:10: fatal error: NvInfer.h: No such file or directory 6 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnxplugin/plugins/HSigmoid.cu:2: src/tensorRT/onnxplugin/onnxplugin.hpp:9:10: fatal error: NvInfer.h: No such file or directory 9 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. Compile CXX src/tensorRT/onnx_parser/onnx2trt_utils.cpp Compile CXX src/tensorRT/onnx_parser/NvOnnxParser.cpp Compile CXX src/tensorRT/onnx_parser/ShapeTensor.cpp Compile CXX src/tensorRT/onnx_parser/ModelImporter.cpp Compile CXX src/tensorRT/onnx_parser/RNNHelpers.cpp Compile CXX src/tensorRT/onnx_parser/onnxErrorRecorder.cpp Compile CXX src/tensorRT/onnx_parser/builtin_op_importers.cpp Compile CXX src/tensorRT/onnx_parser/LoopHelpers.cpp Compile CXX src/tensorRT/onnx_parser/OnnxAttrs.cpp Compile CXX src/tensorRT/onnx_parser/ShapedWeights.cpp Compile CXX src/tensorRT/common/trt_tensor.cpp Compile CXX src/tensorRT/common/cuda_tools.cpp Compile CXX src/tensorRT/common/ilogger.cpp Compile CXX src/tensorRT/common/json.cpp Compile CXX src/tensorRT/infer/trt_infer.cpp Compile CXX src/tensorRT/onnxplugin/onnxplugin.cpp In file included from src/tensorRT/onnx_parser/ShapeTensor.cpp:5: src/tensorRT/onnx_parser/ShapeTensor.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/NvOnnxParser.cpp:5: src/tensorRT/onnx_parser/NvOnnxParser.h:8:10: fatal error: NvInfer.h: No such file or directory 8 | #include "NvInfer.h" | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/onnx2trt_utils.hpp:7, from src/tensorRT/onnx_parser/onnx2trt_utils.cpp:5: src/tensorRT/onnx_parser/ShapedWeights.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/onnx2trt.hpp:7, from src/tensorRT/onnx_parser/ImporterContext.hpp:7, from src/tensorRT/onnx_parser/ModelImporter.hpp:7, from src/tensorRT/onnx_parser/ModelImporter.cpp:5: src/tensorRT/onnx_parser/NvOnnxParser.h:8:10: fatal error: NvInfer.h: No such file or directory 8 | #include "NvInfer.h" | ^~~~~~~~~~~ compilation terminated. Compile CXX src/tensorRT/onnxplugin/plugin_binary_io.cpp In file included from src/tensorRT/onnx_parser/onnxErrorRecorder.cpp:5: src/tensorRT/onnx_parser/onnxErrorRecorder.hpp:7:10: fatal error: NvInferRuntimeCommon.h: No such file or directory 7 | #include "NvInferRuntimeCommon.h" | ^~~~~~~~~~~~~~~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/RNNHelpers.cpp:5: src/tensorRT/onnx_parser/RNNHelpers.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/LoopHelpers.cpp:5: src/tensorRT/onnx_parser/LoopHelpers.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/onnx2trt.hpp:7, from src/tensorRT/onnx_parser/builtin_op_importers.hpp:7, from src/tensorRT/onnx_parser/builtin_op_importers.cpp:5: src/tensorRT/onnx_parser/NvOnnxParser.h:8:10: fatal error: NvInfer.h: No such file or directory 8 | #include "NvInfer.h" | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/OnnxAttrs.cpp:5: src/tensorRT/onnx_parser/OnnxAttrs.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. In file included from src/tensorRT/onnx_parser/ShapedWeights.cpp:5: src/tensorRT/onnx_parser/ShapedWeights.hpp:7:10: fatal error: NvInfer.h: No such file or directory 7 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. make: *** [Makefile:79: objs/tensorRT/onnx_parser/ShapeTensor.cpp.o] Error 1 make: *** Waiting for unfinished jobs.... make: *** [Makefile:79: objs/tensorRT/onnx_parser/onnx2trt_utils.cpp.o] Error 1 make: *** [Makefile:79: objs/tensorRT/onnx_parser/NvOnnxParser.cpp.o] Error 1 make: *** [Makefile:79: objs/tensorRT/onnx_parser/ModelImporter.cpp.o] Error 1 make: *** [Makefile:79: objs/tensorRT/onnx_parser/RNNHelpers.cpp.o] Error 1 make: *** [Makefile:79: objs/tensorRT/onnx_parser/onnxErrorRecorder.cpp.o] Error 1 make: *** [Makefile:79: objs/tensorRT/onnx_parser/builtin_op_importers.cpp.o] Error 1 make: *** [Makefile:79: objs/tensorRT/onnx_parser/LoopHelpers.cpp.o] Error 1 make: *** [Makefile:79: objs/tensorRT/onnx_parser/OnnxAttrs.cpp.o] Error 1 make: *** [Makefile:79: objs/tensorRT/onnx_parser/ShapedWeights.cpp.o] Error 1 In file included from src/tensorRT/onnxplugin/onnxplugin.cpp:2: src/tensorRT/onnxplugin/onnxplugin.hpp:9:10: fatal error: NvInfer.h: No such file or directory 9 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. make: *** [Makefile:81: objs/tensorRT/onnxplugin/onnxplugin.cpp.o] Error 1 src/tensorRT/infer/trt_infer.cpp:6:10: fatal error: NvInfer.h: No such file or directory 6 | #include <NvInfer.h> | ^~~~~~~~~~~ compilation terminated. make: *** [Makefile:81: objs/tensorRT/infer/trt_infer.cpp.o] Error 1
最新发布
07-25
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

_yuki_

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值