使docker-compose支持nvidia-docker方法

1、首先服务器是安装了docker-ce和nvidia-docker2的

使docker-compose支持nvidia-docker方法

然后在安装docker-compose

从官方docker compose版本下载ver1.24.1并授予执行权限

$ curl -L https://github.com/docker/compose/releases/download/1.24.1/docker-compose-`uname -s`-`uname -m` -o /usr/local/bin/docker-compose
$ chmod +x /usr/local/bin/docker-compose

docker-compose version确认 --显示如下安装完成

使docker-compose支持nvidia-docker方法

修改配置文件/etc/docker/daemon.json使docker-compose支持nvidia-docker

修改完成重启docker和daemon,具体如下截图:

使docker-compose支持nvidia-docker方法

编写docker-compose支持的xxx.yaml文件

格式参考如下截图:

使docker-compose支持nvidia-docker方法

使docker-compose支持nvidia-docker方法

简单说明:

services对应下面的nginx服务名,image对应镜像名和版本号,ports对应需要映射的端口,networks对应的网络模式,volumes对应挂载映射的目录或文件

编写完成最后通过执行

docker-compose -f iat-components.yaml up –d   --进行启动

使docker-compose支持nvidia-docker方法

启动完成docker-compose -f iat-components.yaml ps 进行查看

使docker-compose支持nvidia-docker方法

关闭使用docker-compose -f iat-components.yaml down

使docker-compose支持nvidia-docker方法

查看日志:docker-compose -f iat-components.yaml logs 服务名

使docker-compose支持nvidia-docker方法

最后通过nvidia-smi查看识别服务是否调用GPU显卡

使docker-compose支持nvidia-docker方法