转载 https://wxzs5.github.io/2018/06/04/dl-env/
本文为NVIDIA显卡深度学习环境配教程,主要内容为在ubuntu环境下安装NVIDIA显卡驱动、CUDA以及cuDNN。前面如何配置好Ubuntu基本工作环境请参考安装Ubuntu后应该做的事,本文提供一种可行稳妥的安装方法,更详细的安装方式还可参考CUDA官方安装文档与cuDNN官方安装文档。以下操作均在ubuntu 16.04.4 下测试通过,其他发行版本可作参考。

安装NVIDIA显卡驱动
NVIDIA显卡驱动一般有三种安装方法:

使用PPA源安装:最安全,最方便。但是未必有最新驱动,而且下载速度可能慢一点,需要准备好梯子。
安装CUDA时,顺便安装驱动,但未必是最新驱动(不推荐)。
官网下载NVIDIA驱动程序然后本地安装(下载runfile)。
apt-get安装(推荐)
卸载原有驱动
若已有N卡驱动,则需要先卸载:

sudo apt-get remove --purge nvidia*

如果是runfile安装:

sudo chmod +x NVIDIA*.run
sudo ./NVIDIA-Linux-x86_64*.run –uninstall

安装NVIDIA驱动
然后执行:

sudo add-apt-repository ppa:xorg-edgers/ppa #添加ppa源
sudo add-apt-repository ppa:graphics-drivers/ppa #添加ppa源
sudo apt-get update #更新apt-get
sudo apt-get install nvidia-390 #安装390版本的driver

然后可能下载较慢但是最安全,这样安装系统会自动帮你禁用nouveau,所以这样安装完后不用手动禁用nouveau。

runfile 安装
去NVIDIA驱动程序下载官网下载合适的最新驱动,然后本地安装(下载runfile)。
禁用Nouveau驱动

grep nvidia /etc/modprobe.d/* /lib/modprobe.d/*
这个很重要 这个遇到 下面的问题很重要
Unable to load the 'nvidia-drm' kernel module
找这个文件blacklist-nvidia.conf 注释blacklist nvidia-drm
/lib/modprobe.d/blacklist-nvidia.conf:#blacklist nvidia-drm

打开配置文件:

sudo vim /etc/modprobe.d/blacklist.conf

然后在文本最后添加(禁用nouveau第三方驱动,之后也不需要改回来):

blacklist nouveau
options nouveau modeset=0

然后执行:

sudo update-initramfs -u

重启后,执行:

lsmod | grep nouveau
如果没有屏幕输出,说明禁用nouveau成功。

命令行安装驱动
在NVIDIA驱动程序下载官网下载合适的最新驱动,然后按下Ctrl+Alt+F1进入命令行模式,进入驱动所在文件夹,输入以下命令:

sudo service lightdm stop #这会关闭图形界面,但不用紧张
sudo chmod +x NVIDIA-Linux-x86_64-*.run #给驱动run文件赋予执行权限
sudo ./NVIDIA-Linux-x86_64-*.run –no-opengl-files #后面的参数非常重要,不可省略
–no-opengl-files:表示只安装驱动文件,不安装OpenGL文件。这个参数不可省略,否则会导致登陆界面死循环。
–no-x-check:表示安装驱动时不检查X服务,非必需。
–no-nouveau-check:表示安装驱动时不检查nouveau,非必需。
-Z, --disable-nouveau:禁用nouveau。此参数非必需,因为之前已经手动禁用了nouveau。
-A:查看更多高级选项

必选参数解释:因为NVIDIA的驱动默认会安装OpenGL,而Ubuntu的内核本身也有OpenGL、且与GUI显示息息相关,一旦NVIDIA的驱动覆写了OpenGL,在GUI需要动态链接OpenGL库的时候就引起问题。
之后,按照提示安装,成功后重启即可。 如果提示安装失败,不要急着重启电脑,重复以上步骤,多安装几次即可。

测试

nvidia-smi #若列出GPU的信息列表,表示驱动安装成功
nvidia-settings #若弹出设置对话框,亦表示驱动安装成功
image2

安装CUDA
参考官方文档

到CUDA官方下载页面下载CUDA包(runfile)

建议安装版本:cuda 9.0 (tensorflow 现在支持得最好的cuda版本)
然后按照以下步骤安装:

安装
首先确保文件具有运行权限,然后运行:

sudo ./cuda_*linux.run --no-opengl-libs–no-opengl-libs:表示只安装驱动文件,不安装OpenGL文件。必需参数,原因同上。注意:不是-no-opengl-files。
–toolkit:表示只安装CUDA Toolkit,不安装Driver和Samples。
–help:查看更多高级选项。

之后,按照提示安装即可。依次选择:

accept #同意安装
n #不安装Driver,因为已安装最新驱动
y #安装CUDA Toolkit
<Enter> #安装到默认目录
y #创建安装目录的软链接
n #不复制Samples,因为在安装目录下有/samples
CUDA Sample测试
#编译并测试设备 deviceQuery:
cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
sudo make
然后执行:
./deviceQuery

如果测试的最后结果是Result = PASS,说明CUDA安装成功。

把CUDA加到环境变量中

vim ~/.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

安装cuDNN
参考官方文档

到cuDNN官方下载页面下载与CUDA版本相匹配的cuDNN包,要注册一个Nvidia账号,然后才可以下载。
建议的版本:cudnn 7.0 for cuda 9.0

进入安装包所在目录,依次输入以下命令:

tar -xzvf 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
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

至此,关于NVIDIA显卡环境的一切已全部配置完成。

安装 Tensorflow GPU 版本
关于GPU加速训练过程不再强调,推荐tensorflow 1.6 版本,刚好和cuda对应起来,非常稳定。

pip install tensorflow-gpu==1.6 --user #python2
pip3 install tensorflow-gpu==1.6 --user #python3

安装 Pytorch GPU 版本
到Pytorch官网安装最新版本就可以了,pytorch用户体验很好,一般没有兼容性问题。选择你的Python版本和安装方式,推荐使用pip安装,选好cuda 9.0,然后官网会生成安装命令,copy到命令行执行即可安装。

安装 MxNet GPU 版本
参考MxNet官网安装教程安装最新版本就可以了。

安装依赖

sudo apt-get update
sudo apt-get install -y wget python

wget https://bootstrap.pypa.io/get-pip.py && sudo python get-pip.py
安装cuda 9.0 版

pip install mxnet-cu90 --pre --user #python2
pip3 install mxnet-cu90 --pre --user  #python3

其他有一些可选安装项可以看MxNet官网安装选择性安装。
希望能交流更多技术,关注小白的微信公众号吧。

ubuntu 核显输出 独显计算相关推荐

  1. linux自动切换核显和独显,在Ubuntu 18.04上切换独显/集显

    对于笔记本来说,独立显卡太耗电了,非常影响续航:但有时候我们又需要用它来运行一些计算密集型程序.这里推荐两种方法来实现显卡切换. 查看独立显卡是否正在使用 在终端输入 lspci | grep NVI ...

  2. 集显、独显、核显、SOC、POP、微型计算机树莓派的资料

    1.集显.独显.核显的概念与区别: 处理器集成显卡就是指集成在cpu内部的显卡,通常称为核心显卡(也叫APU),如Intel酷睿i3 i5 i7系列处理器以及AMD APU系列处理器中多数都集成了显卡 ...

  3. 电脑核显可以用html吗,核显跟独显能一起用吗 电脑核显和独显一起用,电脑会用哪个...

    核心显卡和独立显卡能同时使用吗 核心显卡是集成在主板上的,独立显卡上可以卸载的: 主板上有核心显卡也是可以安装独立显卡的: 安装独立显卡是自动屏蔽集成显卡性能输出的,没有安装独立显卡驱动就是默认集成显 ...

  4. 台式机核显和独显切换

    机型:成就 5090 核显:Intel® UHD Graphics 630 独显:NVIDIA GeForce RTX 2060 独显 切换 核显 关机 把主机后边的显示器接头从独显上拔下来插到主板上 ...

  5. 大学计算机专业核显和集显,分不清核显和独显?一分钟带你了解清楚

    萌新.小白不管在那个复方都最容易被蒙骗和吃亏,在硬件行业更是如此,因为硬件保质能力远强于一般的产品,这里能做很多文章,之前我们也科普过CPU最容易被蒙骗的地方是分类和命名, 而今天而我们在针对萌新科普 ...

  6. i5集显和独显的区别_集显核显独显有哪些区别 集显核显独显区别介绍【详解】...

    显卡是除处理器之外最受大家关心的硬件装备了,要是显卡配置不到位,其他再好都是浮云,而对于游戏玩家来说,显卡性能差的话更是影响游戏的运行,可折腾人了.今天小编为大家分享"集显".&q ...

  7. i5集显和独显的区别_分不清核显和独显?奸商分分钟坑哭你

    萌新.小白不管在那个复方都最容易被蒙骗和吃亏,在硬件行业更是如此,因为硬件保质能力远强于一般的产品,这里能做很多文章,之前我们也科普过CPU最容易被蒙骗的地方是分类和命名, 而今天而我们在针对萌新科普 ...

  8. 计算机游戏发烧型装配方案,8700k六核1070/1060独显游戏发烧gta组装清单

    i7 7700K升8700k六核1070/1060独显游戏发烧gta组装台式推荐 想要吃鸡不仅需要过硬的技术,还需要有一台性能强劲的电脑,如今硬件价格飞涨,想要5000元配一套高特效吃鸡电脑,那就要在 ...

  9. 计算机专业独显好还是集显好,集显和独显对于普通电脑用户来说区别大吗?

    集显和独显对于普通电脑用户来说区别大吗?相对而言,我们知道独显要比集显图形性能强不少,但价格相对贵不少.对于普通用户来说,电脑是用集显(核显)还是独显好呢?今天我们就聊聊这个话题. 诚然,集显和独显在 ...

最新文章

  1. 骚操作:不重启 JVM,如何替换掉已经加载的类?
  2. pandas将dataframe原有的数据列名称转化为整数数值列名称(convert dataframe column labelsl into integers)
  3. java json解析 代码_Java构造和解析Json数据的两种方法详解一
  4. postgresql根据字符分割数据为多列
  5. 探索Android中的Parcel机制(上) .
  6. iphone看python文件_Python实战 | 只需 ”三步“ 爬取二手iphone手机信息(转发送源码)...
  7. python怎么安装tensorflow-Python使用pip安装TensorFlow模块
  8. P5008-[yLOI2018]锦鲤抄【tarjan】
  9. 领域驱动设计战术模式--领域事件
  10. png格式的图片怎么变成jpg格式?png图片怎么转格式?
  11. c语言四则混合运算可以带括号,带小括号的四则混合运算听后感
  12. Bike Sharing Demand
  13. Win7下安装swoole具体步骤
  14. Chrome浏览器默认全屏启动(非--kiosk模式)
  15. pandas支持读取哪些类型文件_Pandas教程: 读取Excel文件
  16. Maven使用与配置
  17. ChatGPT 辅助专利写作
  18. html5手机号码数据类型,报手机号码的四种格式,你是哪种性格?
  19. Ubuntu手机和电脑安装其他终端Terminal Emulator
  20. spring的ioc和di

热门文章

  1. windows2003玩War3游戏设置
  2. python基础:第一章:元组(四)
  3. 利用C#进行AUTOCAD的二次开发
  4. 比较几种主流PHP开发框架
  5. MySQL 不同条件查询条数 count带条件
  6. 数字化如何给企业赋能
  7. layui 分页的使用
  8. Thinkphp5 实现微信h5 支付
  9. python数据比对怎么做_同事给你迷之Excel数据,4个数据处理案例教你Python数据对比更新...
  10. 黄章深夜多次爆料 魅族16s会比小米9贵但物有所值