安装TensorFlow with GPU

安装TensorFlow with GPU

首先需要安装 NVIDIA CUDA 相关的工具，包括两部分

CUDA® Toolkit 8.0 NVIDIA’s documentation

我选择使用UBUNTU 包管理器的方式安装，根据网站的指示安装，和安装其他的deb没有差别，但是有一个地方出错了。在这一步 6.2.2.2. Compiling the Examples. 错误信息是这样的（抛掉了不重要的信息）: /usr/bin/ld: cannot find -lnvcuvid
解决办法可以参照这个帖子solution:
ruxia

到这一步需要添加的环境变量如下：

• export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
• export LPATH=/usr/lib/nvidia-375:$LPATH
• export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

如果安装的不一样的版本，你需要把 cuda-8.0, nvidia-375 替换成相应的版本

cuDNN v5.1 NVIDIA’s documentation

• 这一步很简单，只需要下载一个压缩包：cuDNN v5.1 Library for Linux.
  然后把这个压缩包解压到你选择的路径，然后把这个路径加到LD_LIBRARY_PATH 就可以了。
• 这个地方有一个错误，我按照网站指示下载cuDNN v5.1P，但是 import tensorflow 的时候出错，错误提示是：libcudnn.so.6 can not found. 所以我就下载了cuDNN v6.0，因为我安装的时候tensorflow 1.3 刚刚发布，所以怀疑是这里的问题。

到这一步需要添加的环境变量如下：

• export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
• export LPATH=/usr/lib/nvidia-375:$LPATH
• export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/home/scott/cuda/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

安装 libcupti-dev library,

这个是 NVIDIA CUDA Profile Tools Interface.

命令如下: sudo apt-get install libcupti-dev

最后安装 tensorflow with gpu

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp35-cp35m-linux_x86_64.whl

最后的URL是对应的python package 的地址，可以通过这个地址找到相应的URL URL of tensorflow python package. 我选择的是 python 3.5 with gpu support.