Installing TensorFlow

TensorFlow
TensorFlow

以 Ubuntu16.04 + GTX1060 为例安装 TensorFlow with GPU support.

CUDA Toolkit 8.0 安装

CUDA(Compute Unified Device Architecture)，是显卡厂商 NVIDIA 推出的运算平台。CUDA™ 是一种由 NVIDIA 推出的通用并行计算架构，该架构使 GPU 能够解决复杂的计算问题。

CUDA 9.0 下载

第一次尝试 Failed. 下载成了 9.0,
TensorFlow 暂时不支持 9.0 (2017/12/09)

CUDA 8.0 下载

cuDNN v6.0

NVIDIA cuDNN 是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN 可以集成到更高级别的机器学习框架中，如加州大学伯克利分校的流行 CAFFE 软件。简单的，插入式设计可以让开发人员专注于设计和实现神经网络模型，而不是调整性能，同时还可以在GPU上实现高性能现代并行计算。

cuDNN v6.0 下载

安装 CUDA 依赖库

$ sudo apt-get install libcuptid-dev

Installing with Anaconda

1. 创建 conda 环境

$ conda create -n tensorflow python=3.6

2. 激活该环境

$ source activate tensorflow

3. 在该环境中安装 TensorFlow

$ pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.4.0-cp36-cp36m-linux_x86_64.whl

验证已经安装完成

运行一个小的 TensorFlow program

1
2
3
4
import tensorflow as tf
hello = tf.constant('Hello TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

分享到