docker-compose.ymlからnvidiaランタイムを指定する方法を教えてください。

Question

私はドッキングウィンドウ-コンからこのコンテナを実行できるようにしたいと思い、次のコマンド/

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

ワット、コマンドラインからGPUへのアクセス/ wのtensorflowコンテナを実行することができています。 フラグをdocker-compose.ymlから指定することはできますか？

Answer 1

/etc/docker/daemon.jsonを編集し、最初のレベルのキー"default-runtime": "nvidia"を追加して、ドッカーデーモン（例："sudo service docker restart"）を再起動します。そのホスト上のすべてのコンテナはnvidiaランタイムで実行されます。