TensorFlow: docker-tensorflow-gpu

code:Dockerfile

FROM tensorflow/tensorflow:latest-gpu

GUIを表示するためのツールをインストール

code:Dockerfile

RUN apt install -y python3-tk

ここで追加のパッケージを入れる

code:Dockerfile

RUN pip3 install matplotlib nptyping

code:docker-compose.yml

version: '2'

services:

tensorflow-gpu:

code:docker-compose.yml

build: .

code:docker-compose.yml

container_name: tensorflow-gpu

GUI用の設定

code:docker-compose.yml

environment:

- DISPLAY=${DISPLAY}

network_mode: 'host'

停止した場合、コンテナを再起動する

code:docker-compose.yml

restart: always

tty: true

code:docker-compose.yml

volumes:

- ./src:/root/src

GUI用の設定

code:docker-compose.yml

- ~/.Xauthority:/root/.Xauthority:ro

code:docker-compose.yml

working_dir: '/root'

code:docker-compose.yml

deploy:

resources:

reservations:

devices:

方法の選択肢が多いが、危険なものもある