使用带有 CUDA 的 Caffe 的 C++ 程序

C++ program using Caffe with CUDA

我尝试 运行 使用 Cuda 编写程序。

构建 Caffe 并进行运行测试很好。

但是我在编译一个简单的代码时得到了错误信息。

当然,我有 link 包含和库。

/usr/local/include/caffe/util/device_alternate.hpp:34:23: fatal error: cublas_v2.h: No such file or directory

你有什么建议吗?

您使用的是哪个版本的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

希望对您有所帮助。