TensorFlow 为 Google 开源的新一代深度学习框架,与之前使用的Caffe和Torch基于层的设计理念有所不同,是基于张量流图的设计思路,这也是今后深度学习框架的一大发展趋势。
本文默认您的linux计算机已经安装有Python下面介绍一下安装与安装遇到的问题及解决方法。将包含下列内容:
CUDA-7.5 与 cuDNN v.5 安装
pip安装
TensorFlow 安装
安装成功测试
安装错误及解决
(一) CUDA-7.5 与 cuDNN v.5 安装
按照NVIDIA官网,分别安装CUDA和 cuDNN ,注意两个版本与计算机显卡都要相对应,下载之前可能需要注册一下,很方便。
(二) pip 安装
在命令行下输入如下安装命令:
$ sudo apt-get install python-pip python-dev
(三) TensorFlow 安装
这部分如果顺利的话一两行代码即可搞定,不顺利的话,就要看人品了。
1. 环境清理
注意:如果你之前在计算机上装过TensorFlow,那么你需要先卸载掉之前的TensorFlow和protobuf,具体如下:
$ pip uninstall tensorflow
$ pip uninstall protobuf
2. 选择合适版本的二进制安装文件链接
# Ubuntu/Linux 64-bit, CPU only, Python 2.7
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.10.0-cp27-none-linux_x86_64.whl
# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
# Requires CUDA toolkit 7.5 and CuDNN v5. For oth