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

举报
Edison 发表于 2019/07/24 20:47:09 2019/07/24
1.2w+ 0 3
【摘要】 配置安装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

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

    全部回复

    上滑加载中

    设置昵称

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

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

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