ubuntu16.04安装gpu版本的tensorflow

时间:2022-04-05 06:30:48

过程参考了http://blog.csdn.net/continueoo/article/details/71404066

安装NVIDIA驱动

禁用自带的nouveau NVIDIA驱动

使用如下命令创建文件

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

文件中添加如下内容:

blacklist nouveau
options nouveau modeset
=0

使用如下命令更新:

sudo update-initramfs -u

重启系统。使用如下命令确认mouveau已经禁用。

lsmod | grep nouveau

输出内容为空。说明成功。

禁用lightdm桌面管理器

sudo /etc/init.d/lightdm stop

使用Ctrl+ Alt + F1进入命令行界面。

安装NVIDIA驱动

切换到下载的NVIDIA驱动的目录下。运行如下命令

sudo ./Your-Nvidia-Driver.run –no-opengl-files

后面这个参数是用来解决重复登录的问题的。

启用lightdm桌面管理器

sudo /etc/init.d/lightdm start

安装CUDA® Toolkit 8.0

下载

从网站https://developer.nvidia.com/cuda-downloads下载合适版本的cuda toolkit。

安装

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

修改环境变量

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}

验证安装成功

cat /proc/driver/nvidia/version

出现如下内容
ubuntu16.04安装gpu版本的tensorflow

nvcc -V

出现如下内容
ubuntu16.04安装gpu版本的tensorflow

其实没有验证完。

安装cuDNN v6

下载

从网站https://developer.nvidia.com/rdp/cudnn-download下载下面要安装的包

安装runtime library

sudo dpkg -i libcudnn6_6.0.21-1+cuda8.0_amd64.deb 

安装developer library

sudo dpkg -i libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb 

安装code samples和cuDNN Library User Guide

sudo dpkg -i libcudnn6-doc_6.0.21-1+cuda8.0_amd64.deb 

验证

cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN

出现test passed信息。

安装libcupti-dev library

sudo apt-get install libcupti-dev

安装gpu版本的tensorflow

sudo pip install tensorflow-gpu

结束。