tensorflow-gpu1.14安装教程

1. 安装CUDA Toolkit

首先，需要安装CUDA Toolkit。可以从NVIDIA官网下载对应版本的CUDA Toolkit，也可以使用以下命令在Ubuntu上安装：

sudo apt-get install nvidia-cuda-toolkit

2. 安装cuDNN

cuDNN是NVIDIA提供的加速深度学习的库。需要从NVIDIA官网下载对应版本的cuDNN，并按照官方文档进行安装。

3. 安装TensorFlow-GPU

可以使用pip安装TensorFlow-GPU：

pip install tensorflow-gpu==1.14

或者从源代码编译安装：

git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
git checkout r1.14
./configure
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
pip install /tmp/tensorflow_pkg/tensorflow-1.14.-cp27-cp27mu-linux_x86_64.whl

4. 测试TensorFlow-GPU

安装完成后，可以使用以下代码测试TensorFlow-GPU是否正常工作：

import tensorflow as tf
tf.test.is_gpu_available()

如果返回True，则表示TensorFlow-GPU已经成功安装并可以使用GPU加速。