经过好几天的折腾终于装好了caffe2,因为我最终目的是要安装Detectron: https://github.com/facebookresearch/Detectron/blob/master/INSTALL.md,所以本文为这个目的服务,是在anaconda2+python2环境下进行的。
参考资料:https://blog.youkuaiyun.com/m0_37644085/article/details/84539523
https://github.com/facebookresearch/Detectron/blob/master/INSTALL.md
https://blog.youkuaiyun.com/zuyuhuo6777/article/details/81450258
安装GPU下的caffe2,需要按照以下步骤进行
1.安装CUDA 9.0, CUDNN 7.0,安装nvcc,这里网上有很多相关博客,不再赘述
2.确保ubuntu的gcc版本是最新版本,我的是5.4.0
3.安装nccl,这里请参考https://blog.youkuaiyun.com/zuyuhuo6777/article/details/81450258
总共有三步:首去官网(https://docs.nvidia.com/deeplearning/sdk/nccl-install-guide/index.html,需要注册一个nvidia账户)下载存储库,然后更新APT数据库,最后执行命令进行安装,官网会给出详细的安装命令。
4.安装opencv3.4.1
5.确保cmake的版本在3.2以上
6.按照caffe2官网指导进行安装,platform选择ubuntu,install type选择"build from source",按照下方给出的指令一步一步进行安装(https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=compile#install-with-gpu-support)
以上每一步骤都绝对不可缺少,否则最后无法安装成功的。