ubuntu16.04安装tensorflow-gpu

ubuntu16.04安装tensorflow

1.anaconda安装 下载地址

cd 下载
bash Anaconda3-5.0.1-Linux-x86_64.sh

一路Enter /yes
记得环境那里填yes,然后source ~/.bashrc就好了

输入anaconda-navigator能打开就配置好了
卸载的话点这

2.CUDA安装

配了最新的环境CUDA9.0+cudnn7.0发现还是一个坑,之前一直是CUDA8.0+cudnn6.0的环境,遇见问题也没记录所以每一次都要重新找,所以这次记录下来.
CUDA8.0下载

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

在 ~/.bashrc 里设置环境变量(不明所以)

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

3.安装cudnn

cudnn6 for cuda8.0linux版的

tar xvzf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

然后安装tensorflow

pip install tensorflow_gpu-1.4.0-cp36-cp36m-manylinux1_x86_64.whl

终端输入python

import tensorflow

没有报错就安装好了
按照这个流程应该不会报错了.