【蜂口 | AI人工智能】搭建caffe依赖环境——龙鹏的一站式caffe工程实践连载(一)

简介: 大家好,我是龙鹏,现在在陌陌深度学习实验室担任资深高级算法工程师,之前在360人工智能研究院担任算法工程师,目前已具有六年的计算机视觉相关的项目经验了。这次的分享我会基于Caffe这个深度学习开源框架,给大家描述一个完整的图像分割项目,大家会对图像分割技术中的基本原理有所了解,并且能够掌握Caffe这个深度学习框架,从而提高自己的应用实战水平。

“ 作者有话说

大家好,我是龙鹏,现在在陌陌深度学习实验室担任资深高级算法工程师,之前在360人工智能研究院担任算法工程师,目前已具有六年的计算机视觉相关的项目经验了。今天来到蜂口,主要是给大家分享一下现在的直播平台,美颜应用中的一些图像技术。

我会结合项目应用,将相关的图像技术的基础知识,和一些细节分享给大家,这次的分享我会基于Caffe这个深度学习开源框架,给大家描述一个完整的图像分割项目,其中包括:数据的准备和处理,Caffe深度学习环境的配置,一个分割网络的设计,Caffe的训练和测试。经过这一次的分享,大家会对图像分割技术中的基本原理有所了解,并且能够掌握Caffe这个深度学习框架,从而提高自己的应用实战水平。 欢迎大家持续关注!“

技术应用场景

当前美容直播行业中广泛使用的图像处理技术,其中像素分割技术是技术难点比较高,应用范围非常广泛的一项技术。有了这门技术之后,我们可以对人脸的各个区域分别进行美颜。比如对嘴唇区域进行美妆,对头发区域进行美发。例如我们看到的美颜相机,背后其实是图像分割技术的应用。

正式开始分享之前,先给大家演示一下,我们这一门技术如何应用在图像直播领域:

https://zhuanlan.zhihu.com/p/44473272

此处由于编辑器的限制,还请通过链接到知乎去看

知识引入

这次我们将分享Caffe的环境配置。本小节的分享内容包含Caffe的依赖环境配置,Caffe的编译,以及其中的一些常见的问题。我们使用的环境是Ubuntu16.04,具体来说包含Nvidia环境配置,主要是Nvidia的显卡驱动的安装,Cuda的环境配置,主要是Cuda的安装。Caffe的环境配置,主要是Caffe的一些依赖Cuda安装,以及在Caffe编译过程中的一些参数的解释。

首先我们开始Nvidia环境配置,第一步是到Nvidia的官网去找到适配于我们的显卡的驱动。

链接地址:https://developer.nvidia.com/cuda-downloads

笔者的电脑显卡是Gt s980,我们可以找到相应的驱动。当然我们也可以直接在Ubuntu下面使用apt-get命令行操作来获取我们对应的显卡驱动。如下是我命令行操作的结果:

sudo ubuntu-drivers devices

image

nvidia-smi命令查看驱动安装结果

image

可以看到命令行获取的显卡驱动与我自己在官网上找到的训练结果是一样的,获取到驱动之后,我们就可以开始进行安装。

驱动安装

首先我们必须进入命令行界面,可以使用Ctrl+Alt+F1命令,关闭掉图形显示的一些东西,然后使用命令就可以进行安装。

安装完之后,我们再重新开启图形界面显示。如果在这个过程中没有发生问题,在安装完驱动之后,我们就可以使用Nvidia-Smi命令,查看驱动安装的结果。

常见问题

第一个问题是我们很多人使用都是笔记本双显卡系统,在安装完驱动之后,就会出现登录界面无限循环,无法进入桌面的情况下,这是为什么呢?这是因为普通的笔记本他一般都包含集显,而Nvidia的显卡都是独显,为了安装独显的驱动,我们会默认我们会关掉Ubuntu默认的驱动。但是如果我们在安装Nvidia驱动的过程中,没有关闭opengl文件的安装,由opengl仍然会去寻找Ubuntu默认的驱动,而只是后者已经被禁掉,这样的话他就因为找不到Ubuntu默认的驱动,而导致启动桌面失败。所以解决问题的办法很简单,我们在安装Nvidia驱动的过程中加上opengl-Files这个选项,这样就可以只安装驱动文件,不安装opengl的文件。

第二个就是Ubuntu默认的驱动。有些设备上会报”一直被占用“这个提示。这个问题就是我们没有禁止Ubuntu默认的驱动。所以解决问题的方法很简单,我们按照下面的步骤去一步一步禁止Ubuntu默认的驱动。

Cuda的环境配置

相对来说,Cuda的环境配置比Nvidia驱动的环境配置要简单得很多。首先我们仍然是到Nvidia的官网上去寻找到Cuda的文件,笔者一般喜欢使用Developer文件,然后就可以使用Nvidia官方提供的命令来进行安装。如下图所示:

image

唯一需要注意的是,因为在前面我们已经安装好了Nvidia驱动,而Cuda他又自带驱动,所以在Cuda安装的最后的一些选项中,选择是否安装驱动这个选项的时候,我们要选择“否”,因为前面已经安装好了驱动,如果在这个时候我们选择的“是”,因为他默认开启的opengl 这个选项,又会导致我们无法进入桌面。在安装完Cuda之后,我们可以使用下面的命令去检验,Cuda是否安装成功。

如果安装Cuda的过程中出现的其他的一些问题,建议大家参考官方的Cuda安装手册。

完整内容及视频解读,请微信搜索:蜂口,关注蜂口小程序~
参与内测,免费获取蜂口所有内容,更有其他优惠福利多多,,若想获得内测种子用户资格,欢迎微信fengkou-IT勾搭,欢迎大家多多参与,尽情挑刺,凡是好的建议,我们都会虚心采纳哒~
蜂口小程序将持续为你带来最新技术的落地方法,欢迎随时关注了解~

相关文章
|
1月前
|
人工智能 自然语言处理 安全
AI创作之旅:探索提示工程的奇妙世界
AI创作之旅:探索提示工程的奇妙世界
17 1
|
1月前
|
人工智能 弹性计算 Serverless
Serverless+AI驱动的一站式数据平台有哪些可能性
【2月更文挑战第4天】Serverless+AI驱动的一站式数据平台有哪些可能性
|
2月前
|
人工智能 数据管理 Serverless
阿里云数据库走向Serverless与AI驱动的一站式数据平台具有重大意义和潜力
阿里云数据库走向Serverless与AI驱动的一站式数据平台具有重大意义和潜力
404 2
|
2月前
|
人工智能 运维 Cloud Native
、你如何看待阿里云数据库走向Serverless与AI驱动的一站式数据平台?
、你如何看待阿里云数据库走向Serverless与AI驱动的一站式数据平台?
149 2
|
2月前
|
人工智能 NoSQL Serverless
基于函数计算3.0 Stable Diffusion Serverless API 的AI艺术字头像生成应用搭建与实践的报告
本文主要分享了自己基于函数计算3.0 Stable Diffusion Serverless API 的AI艺术字头像生成应用搭建与实践的报告
469 6
基于函数计算3.0 Stable Diffusion Serverless API 的AI艺术字头像生成应用搭建与实践的报告
|
3月前
|
机器学习/深度学习 人工智能 自然语言处理
【AI 现况分析】AI 如何提高开发效率,在生产中的实践
【1月更文挑战第27天】【AI 现况分析】AI 如何提高开发效率,在生产中的实践
|
1月前
|
人工智能 JSON 前端开发
【Spring boot实战】Springboot+对话ai模型整体框架+高并发线程机制处理优化+提示词工程效果展示(按照框架自己修改可对接市面上百分之99的模型)
【Spring boot实战】Springboot+对话ai模型整体框架+高并发线程机制处理优化+提示词工程效果展示(按照框架自己修改可对接市面上百分之99的模型)
|
3月前
|
人工智能 Serverless 数据安全/隐私保护
云端服务器应用实践:函数计算X 通义千问快速部署 AI 个人助手应用
云端服务器应用实践:函数计算X 通义千问快速部署 AI 个人助手应用
324 2
|
2月前
|
人工智能 数据管理 大数据
阿里云数据库走向Serverless与AI驱动的一站式数据平台是一个很有前景和意义的发展方向
阿里云数据库走向Serverless与AI驱动的一站式数据平台是一个很有前景和意义的发展方向
33 2
|
1天前
|
人工智能
【AI大模型应用开发】1.2 Prompt Engineering(提示词工程)- 站在巨人的肩膀上,超实用!常用提示词整理
【AI大模型应用开发】1.2 Prompt Engineering(提示词工程)- 站在巨人的肩膀上,超实用!常用提示词整理
3 0