您的位置:首页 > Web前端

ubuntu14.04+cuda8.0+cudnn5.1+caffe+tensorflow 安装教程(三)

2016-12-05 17:51 1066 查看
-要想安装深度学习平台caffe与tensorflow,基础是装好操作系统,装好cuda,再开始装学习平台。教程分为3个部分:

1、ubuntu14.04+cuda8.0+cudnn5.1安装

2、caffe安装

3、tensorflow安装

以此文感谢所有在互联网上分享知识的人和交流群中为我答疑解惑的人。

安装Tensorflow

本文为tensorflow的pip安装,非源码安装。

pip安装很简单,本人之前ubuntu14.04+cuda7.5纠结着源码安装(只有cuda8.0后才可以pip安装,官网上这么说的)。后因nvidia驱动版本的问题,都未安装成功。改成cuda8.0,采用cuda8.0中的nvidia 驱动,再采用pip安装很顺利。

1、祭系统

有些操作装caffe已经做过了,可以不用再做。

sudo apt-get install build-essentil
sudo apt-get update


2、安装pip、python等

sudo apt-get install python-pip
sudo apt-get install python-numpy swig python-dev python-wheel


3、pip安装tensorflow

(1)查看官网源位置并输出源

木有翻墙的话,是不是打不开啊。好吧,暂且给大家贴出来python2.7的GPU和CUP only的源。

# Ubuntu/Linux 64-bit, CPU only, Python 2.7
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.12.0rc0-cp27-none-linux_x86_64.whl


#Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Installing from sources" below.
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-0.12.0rc0-cp27-none-linux_x86_64.whl


(2)pip安装

# Python 2
$ sudo pip install --upgrade $TF_BINARY_URL


结束啦,是不是很简单。通常会要安装很多次。。。因为网的原因。

4、测试

# python
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess =tf.Session()
>>> sess.run(hello)


会输出

'Hello, TensorFlow!'


继续

>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> sess.run(a+b)


会输出

42


大功告成啦!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  caffe tensorflow cudnn