ubuntu16.04安装cuda9.0和GPU版tensorflow(详细)

安装时各个版本之间一定要匹配!可参考如下版本:

ubuntu16.04安装cuda9.0和GPU版tensorflow(详细)_第1张图片

 

1.安装显卡驱动,去官网选择自己合适的显卡

我的电脑适配的显卡是NVIDIA-Linux-x86_64-384.130.run

下载完成后放在home目录下,接着打开终端运行

sudo service lightdm stop

(此时会进入关闭图形界面之后黑屏,此时使用“ctrl+alt+F1”进入命令行界面)

在命令行输入

chmod +x NVIDIA-Linux-x86_64-384.130.run

sudo ./ NVIDIA-Linux-x86_64-384.130.run

(接着会有一系列选项确定安装)

sudo service lightdm start

(返回图形界面)

然后在此处进行修改,选择应用更改选项。

ubuntu16.04安装cuda9.0和GPU版tensorflow(详细)_第2张图片

输入nvidia-smi,测试显卡是否安装成功,如果出现这样的结果即表示安装成功了。

ubuntu16.04安装cuda9.0和GPU版tensorflow(详细)_第3张图片

2.安装cuda

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal

ubuntu16.04安装cuda9.0和GPU版tensorflow(详细)_第4张图片

下载安装包后,在终端输入

sudo sh cuda_9.0.176_384.81_linux.run

ubuntu16.04安装cuda9.0和GPU版tensorflow(详细)_第5张图片

安装提示接着会有一系列选项,在是否安装Linux-x86_64 361.62驱动,这里必须输入“N”不安装否则, cuda后面的安装都会出错。后面的全部按照默认即可,安装完毕后,再声明一下环境变量,并将其写入到 ~/.bashrc 的尾部,终端输入:

echo 'export PATH=/usr/local/cuda-9.0/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc

source ~/.bashrc

sudo reboot

重启后 运行

 

终端输入 nvcc  -V后出现:

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2017 NVIDIA Corporation

Built on Fri_Sep__1_21:08:03_CDT_2017

Cuda compilation tools, release 9.0, V9.0.176

 

说明cuda安装成功。

 

最后测试cuda9.0,在终端执行以下命令:

cd ~/NVIDIA_CUDA-9.0_Samples/1_Utilities/deviceQuery

make

./deviceQuery

输出Result = PASS 表示安装成功。

ubuntu16.04安装cuda9.0和GPU版tensorflow(详细)_第6张图片

 

3.安装tensorflow-gpu_1.8.0

(安装完测试,即可知道需要哪一个版本的cudnn)

pip install tensorflow-gpu==1.8.0

 

完成后输入

python

import tensorflow as tf

报错提示安装缺少libcudnn.so.7,即意味着需要安装cudnn版本为cudnn7.0

ubuntu16.04安装cuda9.0和GPU版tensorflow(详细)_第7张图片

4.安装cudnn

https://developer.nvidia.com/rdp/cudnn-archive

需要注册和登录

ubuntu16.04安装cuda9.0和GPU版tensorflow(详细)_第8张图片

选择点开

Download cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0

点开

cuDNN v7.0.5 Library for Linux

下载 得到

cudnn-9.0-linux-x64-v7.tgz

然后在终端运行

$ tar -zxvf cudnn-9.0-linux-x64-v7.tgz

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/

$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d

$ sudo chmod a+r /usr/local/cuda/include/cudnn.h

$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

 

接着,重新在终端运行

$ nvcc -V

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2017 NVIDIA Corporation

Built on Fri_Sep__1_21:08:03_CDT_2017

Cuda compilation tools, release 9.0, V9.0.176

 

5.再次进行测试

输入

python

import tensorflow as tf

即可看到tensorflow1.8安装成功!

 

 

或者有一种简单的方法,直接在显卡驱动安装好之后,从官网下载anaconda3,因为清华和中科大最近宣布不再提供镜像源,>>>python

可看出python3由annaconda3调用。

>>>conda install tensorflow-gpu==1.8.0会自动在conda环境下安装tensorflow所需要的cuda以及cudnn,比较简单。

 

你可能感兴趣的:(ubuntu16.04安装cuda9.0和GPU版tensorflow(详细))