美文网首页我爱编程
Tensorflow安装教程

Tensorflow安装教程

作者: Mr丶sorrow | 来源:发表于2017-12-05 15:59 被阅读0次

环境

  • GTX1080Ti
  • Ubuntu14.04LTS(阿里的源)

步骤

  1. 安装CUDA8.0
  2. 安装Cudnn
  3. 安装pip
  4. 安装Tensorflow

安装CUDA8.0

总体上的思路参考这篇文章(ubuntu14.04下tensorflow安装指南(GPU)),但是其中坑很多。

  • 下载好安装文件,注意,这个安装文件官方有一版本是错误的,现在已经修复。文件名都是cuda_8.0.61_375.26_linux.run,window下是1.4G大小那个是对的。否则直接安装不了。
  • 安装时,最好不要选择安装opengl,会和ubuntu冲突不能再次打开图形界面了!!!
  • 安装完成后,可能想切换成图形界面,碰到登陆循环的问题,解决方案网上一大堆,但是都没去试。最后我是root下卸载了nvidia的驱动重启就直接成功了。具体原因因为一开始装ubuntu后没有显卡驱动,字都很大模糊,然后先装了驱动。
nvidia-uninstall

坑主要三个,就够受的了。后来问题解决了,找到一篇文章,不错,作者也是用心解决的,不像那些随便抄抄抄的。ubuntu16.04+cuda8.0+cudnn5.1+caffe 安装(基于独立主机&集成显卡&GTX1080Ti)

安装Cudnn

这个好像没什么太多的坑,照着第一个文章走没有什么问题。唯一不确定的是在/usr/local/目录下有两个,一个cuda还有一个cuda-8.0,最后我选择两个都复制了一遍。

安装pip

本身ubuntu就有python2和3,Tensorflow好像基于2的。

sudo apt-get install python-pip
sudo apt-get install python-dev

安装Tensorflow

安装Tensorflow其实很简单,被网上很多的弄得很麻烦,安装网上的一些版本他们都去寻找cuda7.0或者7.5的,所以还是让他自己下载安装吧。对了,网上的那种还要翻墙。

sudo apt-get install tensorflow

测试

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print sess.run(a+b)
42

相关文章

  • Tensorflow安装

    TF安装 Tensorflow windows安装 安装教程可以参考:https://www.tensorflow...

  • TensorFlow

    安装TensorFlow环境 安装教程 https://www.tensorflow.org/install/pi...

  • AttributeError: module 'tensorfl

    根据网上教程安装tensorflow环境时,最后都会有个测试代码用来测试tensorflow是否安装成功 有的朋友...

  • Tensorflow 安装问题解决

    最近在弄机器学习,需要安装tensorflow这个库,于是按照tensorflow中文教程安装,结果很不顺利。接下...

  • Tensorflow Mac 安装教程

    macOS:10.13.4 Tensorflow安装教程在官网或者中文社区也有。也可以参照Tensorflow的G...

  • Tensroflow-GPU环境快速搭建

    现在网上的很多Tensorflow-GPU环境安装教程都是这样的: 安装cuda 安装cudnn 安装tensor...

  • tensorflow安装好找不到models目录问题

    在Tensorflow社区新手教程下安装好tensorflow后,有一个测试是“训练你的第一个Tensorflow...

  • 从conda开始学tensorflow

    星期四, 16. 五月 2019 12:29下午 conda+tensorflow安装教程 本篇博客的安装教程主要...

  • 2020-07-07

    教程用于安装 TensorFlow, 如果你有 Mac 或者 Linux,你不需要这个教程,只需访问TensorF...

  • Tensorflow安装教程

    环境 GTX1080Ti Ubuntu14.04LTS(阿里的源) 步骤 安装CUDA8.0 安装Cudnn 安装...

网友评论

    本文标题:Tensorflow安装教程

    本文链接:https://www.haomeiwen.com/subject/omecixtx.html