热门标签 | 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更新之快,也希望这篇博客对你有帮助。





推荐阅读
  • 精选10款Python框架助力并行与分布式机器学习
    随着神经网络模型的不断深化和复杂化,训练这些模型变得愈发具有挑战性,不仅需要处理大量的权重,还必须克服内存限制等问题。本文将介绍10款优秀的Python框架,帮助开发者高效地实现分布式和并行化的深度学习模型训练。 ... [详细]
  • 本文详细介绍了 Java 网站开发的相关资源和步骤,包括常用网站、开发环境和框架选择。 ... [详细]
  • 本文介绍了如何使用 Google Colab 的免费 GPU 资源进行深度学习应用开发。Google Colab 是一个无需配置即可使用的云端 Jupyter 笔记本环境,支持多种深度学习框架,并且提供免费的 GPU 计算资源。 ... [详细]
  • Python 数据可视化实战指南
    本文详细介绍如何使用 Python 进行数据可视化,涵盖从环境搭建到具体实例的全过程。 ... [详细]
  • 如何高效学习鸿蒙操作系统:开发者指南
    本文探讨了开发者如何更有效地学习鸿蒙操作系统,提供了来自行业专家的建议,包括系统化学习方法、职业规划建议以及具体的开发技巧。 ... [详细]
  • 本文提供了一个使用 while 循环在 Linux Shell 脚本中处理文件列表的具体示例。通过这个例子,读者可以了解如何利用 shell 脚本来批量处理文件,包括文件名的匹配和处理。 ... [详细]
  • 英特尔推出第三代至强可扩展处理器及傲腾持久内存,AI性能显著提升
    英特尔在数据创新峰会上发布了第三代至强可扩展处理器和第二代傲腾持久内存,全面增强AI能力和系统性能。 ... [详细]
  • 兆芯X86 CPU架构的演进与现状(国产CPU系列)
    本文详细介绍了兆芯X86 CPU架构的发展历程,从公司成立背景到关键技术授权,再到具体芯片架构的演进,全面解析了兆芯在国产CPU领域的贡献与挑战。 ... [详细]
  • 最详尽的4K技术科普
    什么是4K?4K是一个分辨率的范畴,即40962160的像素分辨率,一般用于专业设备居多,目前家庭用的设备,如 ... [详细]
  • 使用虚拟机配置服务器
    本文详细介绍了如何使用虚拟机配置服务器,包括购买云服务器的操作步骤、系统默认配置以及相关注意事项。通过这些步骤,您可以高效地配置和管理您的服务器。 ... [详细]
  • 在2019中国国际智能产业博览会上,百度董事长兼CEO李彦宏强调,人工智能应务实推进其在各行业的应用。随后,在“ABC SUMMIT 2019百度云智峰会”上,百度展示了通过“云+AI”推动AI工业化和产业智能化的最新成果。 ... [详细]
  • 在Windows系统中安装TensorFlow GPU版的详细指南与常见问题解决
    在Windows系统中安装TensorFlow GPU版是许多深度学习初学者面临的挑战。本文详细介绍了安装过程中的每一个步骤,并针对常见的问题提供了有效的解决方案。通过本文的指导,读者可以顺利地完成安装并避免常见的陷阱。 ... [详细]
  • 能够感知你情绪状态的智能机器人即将问世 | 科技前沿观察
    本周科技前沿报道了多项重要进展,包括美国多所高校在机器人技术和自动驾驶领域的最新研究成果,以及硅谷大型企业在智能硬件和深度学习技术上的突破性进展。特别值得一提的是,一款能够感知用户情绪状态的智能机器人即将问世,为未来的人机交互带来了全新的可能性。 ... [详细]
  • 深入解析经典卷积神经网络及其实现代码
    深入解析经典卷积神经网络及其实现代码 ... [详细]
  • 本文提供了PyTorch框架中常用的预训练模型的下载链接及详细使用指南,涵盖ResNet、Inception、DenseNet、AlexNet、VGGNet等六大分类模型。每种模型的预训练参数均经过精心调优,适用于多种计算机视觉任务。文章不仅介绍了模型的下载方式,还详细说明了如何在实际项目中高效地加载和使用这些模型,为开发者提供全面的技术支持。 ... [详细]
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社区 版权所有