使用Caffe和CUDA的C++程序

C++ program using Caffe with CUDA

本文关键字:C++ 程序 CUDA Caffe 使用      更新时间:2023-10-16

我试着用Cuda运行一个程序。

制作Caffe和制作runtest很好。

但当我编译一个简单的代码时,我收到了错误消息。

当然,我有链接include和library。

/usr/local/include/caffe/util/device_alternate.hpp:34:23:致命错误:cublas_v2.h:没有这样的文件或目录

你有什么建议吗?

您正在使用哪个版本的Cuda。您可能需要重新安装CUDA。

如果你重新安装cuda后,仍然得到错误:

为了修复cublas错误,你必须包含你的cuda文件夹的路径。

  1. locate cublas_v2.h——这将为您提供cuda安装的路径
  2. export CPLUS_INCLUDE_PATH=/your/cuda/path:$CPLUS_INCLUDE_PATH

您的cuda路径将类似于:/opt/cuda-X.0/include

希望这能有所帮助。