真实机下ubuntu18.04安装NVIDIA驱动,CUDA-9.0,CUDNN-7.05,tensorflow过程(二)

相关链接:
安装NVIDIA驱动步骤
安装tensorflow-gpu步骤

安装CUDA-9.0

若不对gcc进行降级,会出现以下错误:
在这里插入图片描述
(注:高版本可能就不需要降级了,大家以后遇到这样的报错就再降级也可以)

gcc g++降级

测试cuda需要对gcc,g++版本进行降级(降至6.0以下)
ubuntu18.04自带gcc g++ 7.30
查询方式:

gcc --version
g++ --version

下载低版本gcc g++,并进行设置

sudo apt install gcc-4.8 g++-4.8 
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 50 
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.8 50   
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 70 
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 70

这样就可以通过以下命令来切换使用的gcc和g++版本:

sudo update-alternatives --config g++

在这里插入图片描述

sudo update-alternatives --config gcc

在这里插入图片描述

官网下载CUDA

注意:tensorflow对CUDA的版本有要求,见https://github.com/mind/wheels
官网https://developer.nvidia.com/cuda-toolkit-archive下载cuda
根据需求安装适宜版本

在这里插入图片描述

下载文件的列表:

cuda_9.0.176_384.81_linux.run
cuda_9.0.176.1_linux.run
cuda_9.0.176.2_linux.run
cuda_9.0.176.3_linux.run
cuda_9.0.176.4_linux.run

CUDA安装

sudo sh cuda_9.0.176_384.81_linux.run

安装过程中,除了安装NVIDIA显卡的部分选no,其他部分都可以选yes
在这里插入图片描述

安装补丁

sudo sh cuda_9.0.176.1_linux.run
sudo sh cuda_9.0.176.2_linux.run
sudo sh cuda_9.0.176.3_linux.run
sudo sh cuda_9.0.176.4_linux.run

配置环境变量

sudo gedit ~/.bashrc

在文档结尾添加:

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

激活

source  ~/.bashrc

CUDA测试

gcc g++调至低版本

    cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
    sudo make
    ./deviceQuery
    cd ../bandwidthTest
    sudo make
    ./bandwidthTest

测试成功都会出现

Result = PASS

查看cuda版本:

cat /usr/local/cuda/version.txt

安装cudnn

官网下载,本人下载的是cudnn7.0.5 for cuda 9.0

在这里插入图片描述

tar -zxvf cudnn-9.0-linux-x64-v7.tgz
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo cp  cuda/include/cudnn.h /usr/local/cuda/include/
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

安装完毕
安装完成后可用 nvcc -V 命令验证是否安装成功,若出现以下信息则表示安装成功:

yhao@yhao-X550VB:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Tue_Jan_10_13:22:03_CST_2017
Cuda compilation tools, release 8.0, V8.0.61

但可能会找不到nvcc指令,还有一种方式:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

其他方法记载:https://blog.csdn.net/daopper/article/details/81206673针对ubuntu16版本

安装NVIDIA驱动步骤
安装tensorflow-gpu步骤

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值