（docker）进入容器后如何使用本机gpu

首次创建容器，不能直接使用本机gpu

在系统终端进行如下配置：

1.安装NVIDIA Container Toolkit

进入Nvidia官网Installing the NVIDIA Container Toolkit — NVIDIA Container Toolkit 1.15.0 documentation，安装NVIDIA Container Toolkit

本文选用apt安装

2.进入docker文档官网，使用gpu创建并运行容器

 2.1 docker-desktop 

如果是系统安装的是docker-desktop,则进入GPU support in Docker Desktop | Docker Docs

2.2 docker engine

本文安装的是docker engine,进入

Runtime options with Memory, CPUs, and GPUs | Docker Docs

参数解释：

-it: 以交互式方式进入容器（进入容器后可以在终端输入命令）

--rm：退出容器自动删除容器（只有在测试的时候有用，一般不加）

--gpus all: 使用本机所有gpu

ubuntu : 镜像名称

nvidia-smi:执行此命令

上述命令没有指定容器名称，会自动随机生成一个名称，如果要指定名称，可使用参数

--name 容器名

例如：

docker run -it --gpus all --name bev ubuntu

以交互式方式使用ubuntu镜像创建一个名为bev的容器，可使用本机所有gpu

配置一次后，以后创建容器就可以使用本机gpu了

如果不使用本机gpu,就不加--gpus all 创建容器