Ubuntu 18.04 系统安装完成后,我们从零开始搭建基于docker 的深度学习环境。整个安装过程主要分为4步:(1)安装nvidia驱动(2)安装docker (3)安装nvidia docker (4)拉取Deepo镜像。最后,我们通过tensorflow运行一个MNIST手写字符识别的例子来进行验证。
一. 安装nvidia驱动
1. 首先禁用nouveau驱动
把 nouveau 驱动加入黑名单
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
在文件 blacklist-nouveau.conf 中加入如下内容:
blacklist nouveau
options nouveau modeset=0
禁用 nouveau 内核模块
sudo update-initramfs -u
重启
sudo reboot
重启后再次进入命令行
lsmod | grep nouveau
如果没有任何输出说明禁用成功
2. 安装nvidia驱动
需要安装新版本的驱动可以先添加源:
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
然后执行以下命令,列出显卡可用的和推荐的驱动程序
ubuntu-drivers devices