前言
为什么使用docker?
原因是:通过docker运行tensorflow-gpu容器可以避免复杂的环境的配置,比如cuda。具体环境配置相当复杂。。。
而在ubuntu上运行docker tensorflow-gpu镜像,仅仅需要安装nvidia驱动即可,幸运的是一般通过软件更新器(software & updates→ Additional Drivers)即可顺利安装显卡驱动。
docker是一种特别方便的工具,谁用谁知道!
安装
第一步:设置包管理器仓库
在Ubuntu上设置Docker仓库。 lsb_release -cs 可以显示你的 Ubuntu 版本,比如 xenial 或者 trusty。
设置完成后,更新包管理器。
sudo apt-get -y install apt-transport-https ca-certificates
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
sudo apt-get install software-properties-common
sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"
sudo apt-get udate
第二步:安装Docker
sudo apt-get -y install docker-ce
第三步:测试
sudo docker run hello-world
结果如下:
第四部:无需sudo执行Docker命令(非必要)
sudo usermod -aG docker '你的用户名'
到此完成。