热门标签 | HotTags
当前位置:  开发笔记 > 运维 > 正文

Linux系统搭建Tensorflow(GPU加速版)

1.引言纯小白在深度学习服务器上搭建了一次TensorFlow,搭了大概十天左右。深知TensorFlow版本更新之快,国内的一些教程普遍已经有些滞后。如果有条件,建议大家更多翻墙去谷
1.引言
纯小白在深度学习服务器上搭建了一次TensorFlow,搭了大概十天左右。深知TensorFlow版本更新之快,国内的一些教程普遍已经有些滞后。如果有条件,建议大家更多翻墙去谷歌看看TensorFlow的搭建教程,会对自己有很大用处。下面,我们就用TensorFlow最新的版本在深度学习服务器上搭建TensorFlow。PS.笔者所下载时均使用了翻墙软件比如蓝灯,下载速度会非常快,不翻墙的话想要下载非常的麻烦。
2.服务器现有条件(目前配置)
想要安装GPU加速版本的TensorFlow,首先服务器要有GPU。本深度服务器采用的显卡是Nvidia,运行GPU自然需要CUDA和cuDNN支撑。
操作系统:Ubuntu 14.04(目前也有16.04为最新版本)
CUDA版本:8.0

3.相关文件下载
这里我们主要用到cuDNN、Anaconda、TensorFlow的whl文件以及Bazel的文件安装包。由于服务器端网络稳定性不高,在这里,笔者都是将这些安装包下载到本地,然后通过WinSCP(一个传输软件,用法自行百度,很容易)传输到服务器端,以保证安装包的完整性。另外,笔者提醒各位将安装包的上传位置记清楚。

3.1 cuDNN下载

笔者这里所使用的cuDNN的版本为5.1,下载链接为:点击打开链接(需要注册和登录,大家耐心填写吧,毕竟cuDNN还是很关键的)
所下载版本为:cuDNN v5.1 Library for Linux

3.2 Anaconda下载

大多数研究深度学习的人深知深度学习的语言绝大多数还是要使用Python,在这里我们所使用的是Anaconda这个很好用的工具并且Python的版本是2.7
Anaconda下载地址:点击打开链接
此链接中,我们所选择的是Linux系统下的4.3.0,并且是64-bit的Python2.7版本

3.3 TensorFlow 的.whl文件下载

截止今日,笔者所见到的TensorFlow支持GPU版本、Python2.7、CUDA 8.0 以及cuDNN 5. 以上的版本为1.0,(如果有最新版本请尽量使用= =,笔者因为用旧版本不得不翻墙求救才解决-。-)
GPU版本1.0.0 TensorFlow链接如下:点击打开链接

3.4 Bazel 安装包下载

bazel是一种编译指令,需要下载安装包,具体用途后期会有所介绍。
下载路径:点击打开链接 bazel-0.4.4-installer-Linux-x86_64.sh
4. 基于Anaconda的安装

4.1 建立一个conda计算环境(Python 2.7)

安装好Anaconda: bash 指令安装Anaconda。随后在terminal中输入以下指令,创建一个conda的TensorFlow Python2.7环境

安装pip指令

安装TensorFlow
sudo pip install --upgrade TensorFlow所处的位置
激活TensorFlow环境

这个证明TensorFlow已被激活,可以通过 source deactivate 切换出TensorFlow的环境
5. 安装cuDNN
进入服务器cuDNN上传路径,解压文件并进行一系列如下的复制操作

6. 其他依赖安装

6.1 配置CUDA以及cuDNN路径

打开.bash_profile改变环境

添加如下语句:

笔者提醒大家千万别忘了,要用一下语句使该环境生效:


6.2 安装Bazel

6.2.1 安装JDK 8

对于Ubuntu 14.04 操作系统,安装JDK8进行一下一系列操作




其中 PATH_TO_INSTALL.SH为下载的bazel路径的具体目录,至此Bazel指令安装完成
7. 基于源码的TensorFlow安装(GPU加速)
输入以下指令进行TensorFlow的源码下载,默认下载到home中的TensorFlow文件夹中(因为各种原因,很少会一次下载成功,需要有足够耐心,笔者下了大约10次,网速吃紧,各种原因)

7.1 配置TensorFlow(一个大坑)

按照以往的教程,输入以下指令

按照国内各大教程,往往最重要的一步是出现 Do you wish TensorFlow with GPU support?事实上,笔者已发现,在最近更新的TensorFlow版本已经将这句话变为 Do you wish TensorFlow with CUDA support? 因为笔者是小白,对此不了解,栽了一大跟头,希望各位小白也注意哈~到这个问句时记得填Y,其他的都可以填N。但是,CUDA版本和cuDNN版本要清楚标记出,否则后面会报错,很麻烦,笔者深受其害!!其中CUDA 版本填写8.0 cuDNN版本为5.1

7.2创建PIP



8. 测试Tensorflow
笔者提醒大家,因为笔者基于Anaconda使用的TensorFlow GPU加速,所以一定记得激活TensorFlow环境:source activate TensorFlow,后续的测试程序各大教程均适用

9.结束语
TensorFlow可以说是深度学习里面的一个热门框架,很多学习深度学习的研究者都喜欢使用TensorFlow。但是令笔者感到遗憾的是,国内搜索引擎里的各大TensorFlow的教程都有些过时,还是因为TensorFlow更新之快,也希望这篇博客对你有帮助。





推荐阅读
  • 尽管深度学习带来了广泛的应用前景,其训练通常需要强大的计算资源。然而,并非所有开发者都能负担得起高性能服务器或专用硬件。本文探讨了如何在有限的硬件条件下(如ARM CPU)高效运行深度神经网络,特别是通过选择合适的工具和框架来加速模型推理。 ... [详细]
  • 浪潮AI服务器NF5488A5在MLPerf基准测试中刷新多项纪录
    近日,国际权威AI基准测试平台MLPerf发布了最新的推理测试结果,浪潮AI服务器NF5488A5在此次测试中创造了18项性能纪录,显著提升了数据中心AI推理性能。 ... [详细]
  • 如何用GPU服务器运行Python
    如何用GPU服务器运行Python-目录前言一、服务器登录1.1下载安装putty1.2putty远程登录 1.3查看GPU、显卡常用命令1.4Linux常用命令二、 ... [详细]
  • 2017年人工智能领域的十大里程碑事件回顾
    随着2018年的临近,我们一同回顾过去一年中人工智能领域的重要进展。这一年,无论是政策层面的支持,还是技术上的突破,都显示了人工智能发展的迅猛势头。以下是精选的2017年人工智能领域最具影响力的事件。 ... [详细]
  • 吴恩达推出TensorFlow实践课程,Python基础即可入门,四个月掌握核心技能
    量子位报道,deeplearning.ai最新发布了TensorFlow实践课程,适合希望使用TensorFlow开发AI应用的学习者。该课程涵盖机器学习模型构建、图像识别、自然语言处理及时间序列预测等多个方面。 ... [详细]
  • yikesnews第11期:微软Office两个0day和一个提权0day
    点击阅读原文可点击链接根据法国大选被黑客干扰,发送了带漏洞的文档Trumps_Attack_on_Syria_English.docx而此漏洞与ESET&FireEy ... [详细]
  • 本文探讨了亚马逊Go如何通过技术创新推动零售业的发展,以及面临的市场和隐私挑战。同时,介绍了亚马逊最新的‘刷手支付’技术及其潜在影响。 ... [详细]
  • 本文详细介绍了如何在Windows环境下配置GPU支持,并使用Keras和TensorFlow实现YOLOv3模型进行图像目标检测。对于环境搭建的具体步骤,可参考外部链接提供的指南。 ... [详细]
  • 图像分类算法的优化策略与实践
    本文探讨了《Bag of Tricks for Image Classification with Convolutional Neural Networks》论文中的多项技术,旨在通过具体实例和实验验证,提高卷积神经网络在图像分类任务中的性能。文章详细介绍了从模型训练加速、网络结构调整到训练参数优化等多个方面的改进方法。 ... [详细]
  • 随着5G、云计算、人工智能、大数据等新技术的广泛应用,人们的生活生产方式发生了深刻变化。从人际互联到万物互联,数据存储与处理需求激增,推动了数据与算力设施的发展。 ... [详细]
  • 强人工智能时代,区块链的角色与前景
    随着强人工智能的崛起,区块链技术在新的技术生态中扮演着怎样的角色?本文探讨了区块链与强人工智能之间的互补关系及其在未来技术发展中的重要性。 ... [详细]
  • 本文探讨了图像标签的多种分类场景及其在以图搜图技术中的应用,涵盖了从基础理论到实际项目实施的全面解析。 ... [详细]
  • 京东AI创新之路:周伯文解析京东AI战略的独特之处
    2018年4月15日,京东在北京举办了人工智能创新峰会,会上首次公开了京东AI的整体布局和发展方向。此次峰会不仅展示了京东在AI领域的最新成果,还标志着京东AI团队的首次集体亮相。本文将深入探讨京东AI的发展策略及其与BAT等公司的不同之处。 ... [详细]
  • 4000名‘数学基础薄弱’的程序员逆袭成功!揭秘如何学好AI与算法
    对于那些数学基础较差的程序员来说,数学在编程中的重要性往往在工作中才逐渐显现。通过两个实际案例,我们可以深入了解数学如何帮助程序员更好地理解和优化代码。 ... [详细]
  • 本文详细介绍了 TensorFlow 的入门实践,特别是使用 MNIST 数据集进行数字识别的项目。文章首先解析了项目文件结构,并解释了各部分的作用,随后逐步讲解了如何通过 TensorFlow 实现基本的神经网络模型。 ... [详细]
author-avatar
青岛淘宝摄影
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有