如何从docker-compose.yml指定nvidia运行时?

我能够从命令行w /下面的命令运行tensorflow容器瓦特/访问GPU

$ sudo docker run --runtime=nvidia --rm gcr.io/tensorflow/tensorflow:latest-gpu

我想能够从docker-compose运行这个容器。 是否有可能从docker-compose.yml指定--runtime标志?

您应该编辑/etc/docker/daemon.json ,添加第一级密钥"default-runtime": "nvidia" ,重新启动docker守护进程(例如"sudo service docker restart" ),然后该主机上的所有容器都将运行NVIDIA运行时。

更多关于daemon.json 信息