首次创建容器,不能直接使用本机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 创建容器