TensorFlow Engineering with CUDA GPU for Deep Learning

See also:

• TF practical part in do deep learning
• How to setup Docker and Nvidia-Docker 2.0 on Ubuntu 18.04

Install #

Summary: install CUDA first, then TF.
ref TF 1.0 doc
ref nvidia doc, until step 3

requirements #

• 64-bit Linux
• Python 2.7 or 3.3+ (3.5 used in this blog)
• NVIDIA CUDA 7.5 (8.0 if Pascal GPU)
• NVIDIA cuDNN >v4.0 (v5.1 recommended)
• NVIDIA GPU with compute capability >3.0

steps #

1.Manually download “cuDNN v6.0 Library for Linux”.
2.Bash auto Install CUDA in Ubuntu 16.04.2, can combine (&& \) with code below.
3.Install cuDNN, PIP: