Ubuntu 16.04编译GPU(CUDA)版本的Caffe





  • 配置编译环境

参照Ubuntu 14.04,14.10,16.04编译CPU版本Caffe确保可以正常编译通过CPU版本的Caffe.

  • 安装Nvidia CUDA驱动

参照Ubuntu 16.04安装Nvidia CUDA驱动安装Nvidia CUDA驱动.

  • 安装Nvidia CUDNN库

参照Ubuntu 16.04开启dlib对于AVX或者CUDA的支持里面设置Nvidia CUDNN部分,配置好Nvidia CUDNN库.

  • 编译代码

  • 编译出错的处理

如果编译时候出现如下错误信息

则需要在执行cmake的时候增加

的定义。

  • 编译选项

某些库在使用Nvidia CUDNN的时候会崩溃,此时我们可以在编译的时候,排除Nvidia CUDNN即可。

例子如下:

参考链接


ubuntu 16.04中CAFFE配置步骤

发布者

默默

默默码农

发表评论

电子邮件地址不会被公开。 必填项已用*标注