Ubuntu 18.04在已经安装Docker CE后如何安装NVIDIA-Docker,以使docker容器内可以使用宿主机显卡
2023-09-11 14:19:19 时间
注意: 本文是在电脑上已经安装好docker环境的前提下进行的。
docker环境如何安装参照前文。
本文内容节选自: https://baimafujinji.blog.csdn.net/article/details/89784555
---------------------------------------
安装NVIDIA-Docker
# Add the package repositories curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \ sudo apt-key add - distribution=$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \ sudo tee /etc/apt/sources.list.d/nvidia-docker.list sudo apt-get update
正式安装NVIDIA-Docker:
# Install nvidia-docker2 and reload the Docker daemon configuration sudo apt-get install -y nvidia-docker2 sudo pkill -SIGHUP dockerd
用最新的CUDA镜像来测试一下nvidia-smi(检查是否安装成功,安装成功,则会显示关于GPU的信息)。
# Test nvidia-smi with the latest official CUDA image docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi
如下图所示:
三、用Docker(NVIDIA-Docker)载入TensorFlow镜像
参考Dockerhub关于Tensorflow的页面(链接),主要是“Running Containers”部分。例如,如果要开启一个基于Python2、CPU版的TensorFlow,可以使用:
docker run -it --rm tensorflow/tensorflow bash
或者,开启一个基于Python3、GPU版的TensorFlow(可根据需要的版本指定镜像),并开启容器。注意,这里我们使用的Tag是1.10.0-gpu-py3。因为CUDA的版本是9.0,太高版本的TensorFlow不支持。更多的可选Tag可以查阅链接。
docker run -it --rm --runtime=nvidia tensorflow/tensorflow:1.10.0-gpu-py3 python
来看一下执行的结果。
参考资料:
相关文章
- Docker的ubuntu镜像安装的容器无ifconfig和ping命令的解决
- docker脚本自动化安装
- Ubuntu18.04-docker-compose安装
- docker探索-CentOS7中配置Docker的yum源并升级安装docker1.13(十)
- 【Docker】docker安装Jenkins
- 【Docker】docker的安装和常用命令
- Ubuntu安装docker
- Docker - 在Ubuntu16.04中安装Docker CE
- fedora30平台安装docker 19.03
- docker安装最新版Jenkins:拉取镜像/创建容器
- Docker的三种网络代理配置:dockerd pull镜像代理;容器docker run网络代理;docker build代理--build-arg
- yum安装docker(阿里镜像源)及docker-compose二进制安装
- Docker cAdvisor安装
- Docker-compose 安装 rocketmq
- Docker与iptables 只允许特定ip访问Docker的服务 通过iptables限制docker容器端口
- 018-docker-docker-compose定义和运行多容器 Docker 应用程序的工具
- 006-docker-安装-nginx
- 004-docker命令-docker资源配置-Docker容器CPU、memory资源限制
- 77.第十七章 企业级容器技术docker -- Docker 镜像制作和管理(五)
- 76.第十七章 企业级容器技术docker -- Docker 镜像制作和管理(四)
- 一行命令实现在Docker镜像容器中安装JDK11
- docker 权限问题 Got permission denied while trying to connect to the Docker daemon socket at 。。。
- Docker Ubuntu容器安装ping
- SonarQube----使用 docker-compose 安装部署SonarQube
- 结合实践总结Docker安装Nacos
- Docker学习笔记17:docker实例之安装 Node.js、PHP、MySQL、Tomcat、Python、Redis、MongoDB、Apache
- Docker学习笔记13:docker使用之Compose
- Docker学习笔记07:docker使用之容器使用
- 完成docker安装centos7操作系统实战
- docker中使用tomcat