注意:CUDA自带显卡驱动,不用另装

1.进入tensorflow安装界面https://www.tensorflow.org/install/install_linux查看要求(需要v*n,不是必须)

UBUNTU 16.04深度学习机环境搭建(三)——CUDA9.0+CUDNN7.0+tensorflow1.7安装

2.安装CUDA

进入cuda documentation http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4VZnqTJ2A

UBUNTU 16.04深度学习机环境搭建(三)——CUDA9.0+CUDNN7.0+tensorflow1.7安装

点击右上角older https://developer.nvidia.com/cuda-toolkit-archive

选择想要安装的版本,设置系统(这里是9.0的安装命令,官方有错)

wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb

sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub

sudo apt-get update

sudo apt-get install cuda

3.安装CUDNN

在https://developer.nvidia.com/rdp/cudnn-download下载CUDNN(需要NVIDIA账号)

tar -xzvf cudnn-9.0-linux-x64-v7.solitairetheme8

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*

4.设置环境变量


~/.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

5.安装tensorflow-gpu

sudo pip install tensorflow-gpu

相关文章: