一,首先显卡更新为nvidia
1,先程序搜索更新
$ sudo apt-get update
2,去设置——软件和更新中的附加驱动选择nvidia驱动
二,安装CUDA
$ sudo dpkg -i cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda
三,安装cudnn
$ 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*
五,配置环境变量
$ sudo gedit ~/.bashrc
加入以下内容
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
命令行中执行
source ~/.bashrc
六,pip安装tensorflow_gpu_1.4.0
$ sudo pip3 install tensorflow_gpu-1.4.0-cp34-cp34m-manylinux1_x86_64.whl
七,测试安装的tensorflow
至此tensorflow 安装教程结束