码迷,mobileinhere.cn
首页 > 系统相关 > 详细

ubuntu 18.04上cuda 9.0、cudnn7.0及tensorflow 1.8的安装

时间:2018-05-08 14:39:24      阅读:2166      评论:0      收藏:0      [点我收藏+]

标签:sudo   符号链接   ref   不安装   uda   出现   http   pip源   官网   

配置

笔者使用dell inspiron 7559笔记本电脑,显卡为nvidia gtx 960m。

目标

由于本机显卡仅有nvidia-384驱动包能够良好支持(nvidia-387、nvidia-390包均在本机出现了系统无法登陆等异常),而cuda 9.1需要驱动至少为nvidia-387,故选择安装cuda 9.0及cudnn7.0。
telsorflow 1.8完全支持cuda 9.0因此可以使用最新版。

安装显卡驱动

使用apt安装nvidia-384驱动包(实际安装驱动为390):

sudo apt-get install nvidia-384

安装完成后重启系统。
运行命令

nvidia-smi

若能够显示显卡当前运行状况(表格)则代表驱动安装成功。

安装cuda 9.0

准备工作

cuda及tensorflow体积均较大且下载源在国外,请找到一个良好的网络环境。可以寻找适当的国内镜像。
从nvidia官网legacy releases下载cuda 9.0版本的.run安装包(developer.nvidia.com/cuda-90-download-archive)。
由于cuda 9.0仅支持gcc 6.0及以下版本,而ubuntu 18.04预装gcc版本为7.3,故手动安装gcc-6与g++-6:

sudo apt-get install gcc-6 g++-6

之后切换至/usr/bin目录修改符号链接,使gcc 6成为默认使用版本:

cd /usr/bin
sudo rm gcc
sudo ln -s gcc-6 gcc
sudo rm g++
sudo ln -s g++-6 g++

安装

为cuda 9.0安装包赋予运行权限并运行之:

chmod 775 cuda_9.0.176_384.81_linux.run
sudo ./cuda_9.0.176_384.81_linux.run --no-opengl-libs

安装过程中选择不安装驱动,仅安装cuda与samples。若提示"you are attempting to install on an unsupported configuration."选择y强制安装(如果前面修改了gcc版本,此处应不报该错误)。
安装完成后若提示安装失败,可参考错误提示增加--override参数重新安装。

测试

进入samples所在目录(默认为~/nvidia_cuda-9.0_samples),运行命令(所需时间较长)

make

(若不愿等候太长时间也可进入子目录单独make一些示例程序)
若编译成功(可能有warning)则可以进入bin目录运行其中的程序。以上皆成功时,则cuda安装成功。

安装cudnn 7.0

仍从nvidia官网下载cudnn 7.0,注意需要注册nvidia账户才能下载。
之后正常安装其所有包(笔者使用的是.deb包)

sudo dpkg -i libcudnn7_7.0.4.31-1+cuda9.0_amd64.deb libcudnn7-dev_7.0.4.31-1+cuda9.0_amd64.deb libcudnn7-doc_7.0.4.31-1+cuda9.0_amd64.deb

安装tensorflow 1.8

使用pip安装。若无pip则首先安装pip:

sudo apt-get install python3-pip
#若python2则为
#sudo apt-get install python-pip

之后安装tensorflow:
由于pip官方源下载速度较慢,可以参考网上办法修改pip源至国内镜像,例如清华大学tuna镜像

sudo pip3 install tf-nightly-gpu

安装完成后运行tf示例程序,检查tensorflow是否顺利运行。

最后的强迫症

以上步骤均完成后,可将默认gcc版本改回7.3,经笔者实测,应不影响tensorflow的正常使用。
命令:

cd /usr/bin
sudo rm gcc
sudo ln -s gcc-7 gcc
sudo rm g++
sudo ln -s g++-7 g++

ubuntu 18.04上cuda 9.0、cudnn7.0及tensorflow 1.8的安装

标签:sudo   符号链接   ref   不安装   uda   出现   http   pip源   官网   

原文地址:www.cnblogs.com/arrowkeys/p/9007437.html

(0)
(0)
   
举报
评论 一句话评论(0
0条  
登录后才能评论!
2014 mobileinhere.cn 版权所有 京icp备13008772号-2
华人娱乐注册