使用带有 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 文件夹的路径。
locate cublas_v2.h
-- 这应该给你 cuda 安装的路径
export CPLUS_INCLUDE_PATH=/your/cuda/path:$CPLUS_INCLUDE_PATH
您的 cuda 路径类似于:/opt/cuda-X.0/include
希望对您有所帮助。
我尝试 运行 使用 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 文件夹的路径。
locate cublas_v2.h
-- 这应该给你 cuda 安装的路径export CPLUS_INCLUDE_PATH=/your/cuda/path:$CPLUS_INCLUDE_PATH
您的 cuda 路径类似于:/opt/cuda-X.0/include
希望对您有所帮助。