【Docker】 容器内如何使用GPU资源

Docker container 内使用GPU

Author：Xin Pan

Date: 2022.08.16

---

也正好是最近在使用Docker 做一些部署和开发的工作，之前用的比较基本，但是现在需要做一些更深入的使用。其中就是GPU在容器内的的配置，以及如何在container内使用GPU进行计算。

当我们使用docker 创建一个container的时候，可能需要使用GPU资源。那怎么做呢。

下边的命令是使用所有的GPU资源

docker run -it --gpus all -v out_path:container_inner_path -p out_port:container_inner_port --shm-size=32g --name container_name image_name:tag bash

当我只想指定某一个GPU资源比如，id=3的卡时，命令就是这样的。

docker run -it --gpus "device=3" -v out_path:container_inner_path -p out_port:container_inner_port --shm-size=32g --name container_name image_name:tag bash