转载 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,

本文详述了在Ubuntu 16.04环境下安装NVIDIA显卡驱动、CUDA 9.0和cuDNN 7.0的步骤,包括apt-get安装驱动、禁用Nouveau、安装CUDA及cuDNN,并提供了Tensorflow、Pytorch和MxNet的GPU版本安装指南。

4139

被折叠的 条评论
为什么被折叠?



