设为首页 收藏本站
查看: 1160|回复: 0

[经验分享] Keras学习环境配置-GPU加速版(Ubuntu 16.04 + CUDA8.0 + cuDNN6.0 + Tensorflow)

[复制链接]

尚未签到

发表于 2017-11-19 21:33:35 | 显示全部楼层 |阅读模式
          本文是个人对Keras深度学习框架配置的总结,不周之处请指出,谢谢!
          1. 首先,我们需要安装Ubuntu操作系统(Windows下也行),这里使用Ubuntu16.04版本:
   DSC0000.png
          2. 安装好Ubuntu16.04之后,需要对系统进行初始化设置及更新:
          打开终端输入:
     系统升级:

          →~ sudo apt-get update
          →~ sudo apt-get upgrade
          安装基础依赖库:
          →~ sudo apt-get install python-dev python-pip python-nose gcc g++ git gfortran libopenblas-dev liblapack-dev libatlas-base-dev
          →~ sudo apt-get install setuptools wheel python-numpy python-scipy python-matplotlib
          3. 安装CUDA开发环境
          下载CUDA8.0:

   DSC0001.png
          打开终端,切换到下载目录:
          →~ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
          →~ sudo apt-get update
          →~ sudo apt-get install cuda
          安装完成后,配置CUDA路径:
          →~ sudo gedit /etc/profile
          在profile文件末尾回车添加(注意:如果使用的不是8.0版本需要修改版本号):
          →~ export CUDA_HOME=/usr/local/cuda-8.0
          →~ export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
          →~ export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
          修改之后:
          →~ source /etc/profile
          验证是否配置成功:
          →~ nvcc -V
          出现下面的信息即成功:
   DSC0002.png
          4. 安装cuDNN加速库
          本文采用的是CUDA8.0,对应的安装cudnn-8.0-linux-x64-v6.0.tgz。下载解压出来是名为cuda的文件夹,里面有bin、include、lib,将三个文件夹复制到安装CUDA的地方覆盖对应文件夹:
          →~ cd /home/..../cudnn/
          →~ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
        →~ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
          →~ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
          5. 安装Keras框架
          →~ sudo pip install scikit-learn scikit-image
          →~ sudo pip install tensorflow-gpu  # GPU加速版
          →~ sudo pip install keras
          在终端中验证是否安装成功:
          →~ import tensorflow
          →~ import keras
          如果不报错,即配置成功!
  

运维网声明 1、欢迎大家加入本站运维交流群:群②:261659950 群⑤:202807635 群⑦870801961 群⑧679858003
2、本站所有主题由该帖子作者发表,该帖子作者与运维网享有帖子相关版权
3、所有作品的著作权均归原作者享有,请您和我们一样尊重他人的著作权等合法权益。如果您对作品感到满意,请购买正版
4、禁止制作、复制、发布和传播具有反动、淫秽、色情、暴力、凶杀等内容的信息,一经发现立即删除。若您因此触犯法律,一切后果自负,我们对此不承担任何责任
5、所有资源均系网友上传或者通过网络收集,我们仅提供一个展示、介绍、观摩学习的平台,我们不对其内容的准确性、可靠性、正当性、安全性、合法性等负责,亦不承担任何法律责任
6、所有作品仅供您个人学习、研究或欣赏,不得用于商业或者其他用途,否则,一切后果均由您自己承担,我们对此不承担任何法律责任
7、如涉及侵犯版权等问题,请您及时通知我们,我们将立即采取措施予以解决
8、联系人Email:admin@iyunv.com 网址:www.yunweiku.com

所有资源均系网友上传或者通过网络收集,我们仅提供一个展示、介绍、观摩学习的平台,我们不对其承担任何法律责任,如涉及侵犯版权等问题,请您及时通知我们,我们将立即处理,联系人Email:kefu@iyunv.com,QQ:1061981298 本贴地址:https://www.yunweiku.com/thread-408672-1-1.html 上篇帖子: solr的访问权限管理及ubuntu下iptables的设置 下篇帖子: VMware Workstation Pro 12 创建虚拟机(安装Ubuntu)
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

扫码加入运维网微信交流群X

扫码加入运维网微信交流群

扫描二维码加入运维网微信交流群,最新一手资源尽在官方微信交流群!快快加入我们吧...

扫描微信二维码查看详情

客服E-mail:kefu@iyunv.com 客服QQ:1061981298


QQ群⑦:运维网交流群⑦ QQ群⑧:运维网交流群⑧ k8s群:运维网kubernetes交流群


提醒:禁止发布任何违反国家法律、法规的言论与图片等内容;本站内容均来自个人观点与网络等信息,非本站认同之观点.


本站大部分资源是网友从网上搜集分享而来,其版权均归原作者及其网站所有,我们尊重他人的合法权益,如有内容侵犯您的合法权益,请及时与我们联系进行核实删除!



合作伙伴: 青云cloud

快速回复 返回顶部 返回列表