-要想安装深度学习平台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
大功告成啦!