简单的方法：在 Ubuntu 18.04 上安装 Nvidia 驱动程序、CUDA、CUDNN 和 Tensorflow GPU

Question

如何轻松运行 tensorflow gpu？

Answer 1

我刚刚在 Ubuntu 18.04 上安装了 Tensorflow GPU。有很多关于它的说明，但是我认为通常不会使用最快和最简单的方法，我想分享它：

英伟达驱动：

ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
nvidia-smi

CUDA：

通常：“sudo apt install nvidia-cuda-toolkit”但是这会安装 9.1 版，目前太新，tensorflow 将无法运行。而是下载 CUDA 9.0：https : //developer.nvidia.com/cuda-90-download-archive? target_os = Linux&target_arch = x86_64&target_distro = Ubuntu&target_version = 1704&target_type =runfilelocal 然后运行（第一行的其他详细信息可以在以下位置找到：How can我在 Ubuntu 17.10 上安装了 CUDA 9）：

sudo sh cuda_9.0.176_384.81_linux.run --override
sudo ln -s /usr/bin/gcc-6 /usr/local/cuda/bin/gcc
sudo ln -s /usr/bin/g++-6 /usr/local/cuda/bin/g++
cd ~
sudo nano .bashrc

add at the end of the file:
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
CTL+X to save and exit

神经网络：

在 nvidia 开发人员处注册https://developer.nvidia.com/cudnn 下载 16.04 的 9.1 运行时和开发人员库（文件 cuDNN v7.1.3 运行时库适用于 Ubuntu16.04（Deb）和 cuDNN v7.1.3 开发人员库适用于 Ubuntu16.04（ Deb)) 用软件管理器打开文件并安装它们。检查：

cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2

安装库和 tensorflow：

sudo apt-get install libcupti-dev
pip3 install tensorflow-gpu

查看：

in tensorflow check for GPU support 
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

注意：从终端启动 tensorflow 或您的开发环境，否则对我来说它不会加载 PATH 变量。