caffe cuda docker 环境搭建

最新推荐文章于 2024-07-26 20:28:45 发布

栾金鹿

最新推荐文章于 2024-07-26 20:28:45 发布

阅读量583

点赞数

分类专栏： deep learning 文章标签：深度学习 cuda nvidia

本文链接：https://blog.csdn.net/luanjinlu/article/details/79087163

版权

本文介绍了如何在CUDA 9.0和CUDNN 7.0环境下搭建Caffe的Docker环境。重点包括理解CUDA、CUDNN和Docker的关系，确定GPU驱动版本，安装NVIDIA驱动，选择正确的Docker镜像，以及编译和构建定制的Caffe Docker镜像。过程中强调了版本一致性的重要性以及遇到问题时的解决方法。

摘要由CSDN通过智能技术生成

注意：

在安装过程中，要时刻提醒自己注意版本的一致性(若不一致，会走很多弯路)，包括nvidia driver cuda cudnn 和caffe ，我们需要的版本是： dirver:384以上 cuda:9.0 cudnn:7.0 caffe:1.0

几个概念

Cuda是nvidia的驱动模块，用来做gpu并发计算，它提供内核驱动和应用层的工具nvidia-smi；
Cudnn，是nvidia为深度学习封装的SDK，提供给那些使用nvidia GPU的应用统一的接口，比如caffe；
那么，nvidia为了提供简易的安装，对cuda和cudnn都提供了docker的支持，并且不断的更新维护，

开始

好了，到了我们说说caffe的时候了，Caffe是深度学习的框架，跟tensfllow属于一个地位的，caffe可以基于cpu／gpu来进行学习计算，那么，当caffe 使用gpu进行运算的时候，就用到了nvidia的gpu，也就意味着需要用到 cuda和cudnn。当然，cudnn并不是必须的，所以在编译caffe时，可以enable cudnn，也可以disable cudnn，

OK，我们再来说说caffe的版本维护，它是有BVLC进行维护的，并且BVLC提供docker的版本 BVLC／caffe，到现在，我们搞清楚了前后联系，那么在安装和参考官方文档的时候就不会出现逻辑混乱了。

现在，