ubuntu18.04部署CUDA10.0+cuDNN+tensorflow-gpu+docker

举报
Edison 发表于 2019/07/24 20:47:09 2019/07/24
【摘要】 配置安装nvidia GPU环境

关闭nouveau

更新ubuntu系统

apt-get update

apt-get upgrade

 

新建blacklist-nouveau.conf文件

vi /etc/modprobe.d/blacklist-nouveau.conf

增加以下内容:

blacklist nouveau

blacklist lbm-nouveau

options nouveau modeset=0

alias nouveau off

alias lbm-nouveau off

重启:

reboot



      下载相关CUDA deb包


官网链接:

Cuda:

https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=deblocal


图片 1.png


cuDNN:

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

图片 2.png

查看显卡信息

 lspci | grep -i vga

 

安装PPA源

sudo add-apt-repository ppa:graphics-drivers/ppa

 

查找适用于本机GPU的nvidia-dirver版本:

https://www.nvidia.com/Download/index.aspx?lang=en-us

图片 3.png


查询nvidia驱动可用版本,运行 sudo apt-cache search nvidia

apt-get install nvidia-utils-410  安装相应版本

 

逐步安装

nvidia-smi 显示GPU运行信息:

图片 4.png

nvcc -V 显示Cuda版本信息:

图片 5.png













【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。