搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV

前言

该文章是之前作的云上深度学习环境搭建笔记,当时也花了很多心血,在面试助攻手册的整理发布过程当中穿插着来一篇,放松一下。若对机器学习感兴趣的小伙伴,能够亲自上手体验一番,颇有乐趣。python

因为科研任务,须要在云端运行一个基于神经网络的目标识别库,须要用到GPU加速。亚马逊有不少自带GPU的机器,可是环境的配置可折腾坏了,尤为是opencv,每次总会出各类各样的问题!git

无奈中,看见了Adrian Rosebrock的英文博文:Pre-configured Amazon AWS deep learning AMI with Pythongithub

其已经预配置好各类环境,包括Keras, TensorFlow, scikit-learn, scikit-image, OpenCV等。能够说是很棒棒喽。面试

此文章并不是其文章的翻译稿,而是提炼主要内容而且亲自实践而来,补充了不少坑。如条件容许,还请自行阅读英文原文,更详细。算法

第一步:竞价获取服务器

注册亚马逊什么的就不说了,注意免费一年政策并不能用于这些带GPU的机型就是了。ubuntu

使用竞价请求来申请主机,至于为何要用竞价实例,第一是便宜,第二是我发现个人帐号没有申请持续性gpu主机实例的权限。至于竞价实例是什么,请google/baidu。安全

搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV

原文中,给了四种适合的机型:服务器

1c4.xlarge: $0.199/hour
2p2.xlarge: $0.90/hour
3p2.8xlarge: $7.20/hour
4p2.16xlarge: $14.40/hour

For CPU instances I recommend you use the “Compute optimized” c4.* instances. In particular, the c4.xlarge instance is a good option to get your feet wet. If you would like to use a GPU, I would highly recommend the “GPU compute” instances. The p2.xlarge instance has a single NVIDIA K80 (12GB of memory). The p2.8xlarge sports 8 GPUs. While the p2.16xlarge has 16 GPUs.微信

然而,我选择g2,8cpu,15g ram,带一块GPU,最基础的任务足够。网络

如图选择,搜索ami-ccba4ab4,找到这一个预配置的AMI镜像
搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV

请注意这里本来是16G的卷,建议最好20-30G,防止你还要配置或者安装些东西,以前没用这个镜像,随便编译个opencv,硬盘就不够了。

搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV
提交!接下来即是等待,有快有慢,快的时候马上就有机器。

搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV

第二步:一键安装显卡驱动

拿到机器后,使用ssh链接上,用户名是ubuntu。
搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV

这里有个坑就是显卡驱动。原文已经说了,尽管镜像中原本安装好了驱动,因为挂载盘特性,驱动会可能被系统屏蔽,须要从新安装,或者是屏蔽某系统文件(没仔细研究,我以为重装来得快,哈哈哈)。

首先,检查是否有驱动:

1nvidia-smi

若是显示:

搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV
则已经有驱动了,不须要安装。

不然,你须要安装驱动,做者已经在/installers下放了显卡驱动。安装:

1cd installers
2sudo ./NVIDIA-Linux-x86_64-375.26.run --silent

值得注意的是,我用的g2这个机器的显卡K520,是不能够用这个驱动的,咱们须要从新下载驱动,下载地址以下

http://www.nvidia.com/download/driverResults.aspx/108586/en-us

供你们核对信息:

搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV

将下载的驱动也放在该文件夹后,先要给该文件执行的权限,而后静默安装:

1sudo chmod 777 NVIDIA-Linux-x86_64-367.57.run
2sudo ./NVIDIA-Linux-x86_64-367.57.run --silent

安装时会弹出一些警告,不用理会,安装好后,再次执行nvidia-smi,获得以前的图。

如今,你的环境已经配置好了。

第三步:开启python虚拟环境

最后一步,须要打开大神预先配置好的环境,执行

1workon dl4cv

搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV
完成!你可使用pip-freeze查看下:

搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV
最后,放一张运行我程序的图:
搭建AWS云服务器深度学习环境免环境配置/GPU支持/Keras/TensorFlow/OpenCV

若有问题,能够留言讨论!

关注我

我是蛮三刀把刀,目前为后台开发工程师。主要关注后台开发,网络安全,Python爬虫等技术。

来微信和我聊聊:yangzd1102

Github:https://github.com/qqxx6661

原创博客主要内容

  • 笔试面试复习知识点手册
  • Leetcode算法题解析(前150题)
  • 剑指offer算法题解析
  • Python爬虫相关技术分析和实战
  • 后台开发相关技术分析和实战
    同步更新如下博客
  1. Csdn

http://blog.csdn.net/qqxx6661

拥有专栏:Leetcode题解(Java/Python)、Python爬虫开发

  1. 知乎

https://www.zhihu.com/people/yang-zhen-dong-1/

拥有专栏:码农面试助攻手册

  1. 掘金

https://juejin.im/user/5b48015ce51d45191462ba55

  1. 简书

https://www.jianshu.com/u/b5f225ca2376

我的公众号:Rude3Knife

若是文章对你有帮助,不妨收藏起来并转发给您的朋友们~

相关文章
相关标签/搜索