ubuntu安装tensorflow gpu版本

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/owenfy/article/details/81674661

由于之前用的1.4，cuda8.0，所以这次重装了一下、

先装驱动

再装cuda9.0，下载

sudo sh cuda_9.2.148_396.37_linux.run

Do you accept the previously read EULA，  输入 accept，进入下一步。
INSTALL NVIDIA Accelerates Graphics Driver for Linux-x86_64 375.26，  输入n，进入下一步；
Install the  CUDA 9.0 Toolkit?，  输入y， 进入下一步；
Enter Tookit Location 点击enter， 进入下一步；
Do you want to install a symbolic link at /usr/local/cuda，  输入y，进入下一步；
Install the CUDA 9.0 Samples?  输入y， 进入下一步；
Enter CUDA Samples Location 点击enter， 进入下一步；

内容

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

   

成功

下载cudann

有个文件是不tgz格式

cp cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz

tar -xvf cudnn-8.0-linux-x64-v5.1.tgz

安装照着官网来

最后照着官网来装tensorflow pip一下就好