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

Question

我尝试使用 Cuda 运行程序。

构建 Caffe 并进行 runtest 很好。

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

当然，我有链接包含和库。

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

你有什么建议吗？

Answer 1

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

希望这可以帮助。