Docker から NVIDIA GPU が見えるか確かめるスニペット
docker compose up するとコンテナから GPU が見えるか確認できる
code:docker-compose.yml
version: '3.8'
services:
smi:
image: nvidia/cuda:11.1-base
command: nvidia-smi
network_mode: none
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities:
- gpu
tf:
image: tensorflow/tensorflow:latest-gpu
command: python -c "import tensorflow as tf; tf.test.gpu_device_name()"
network_mode: none
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities:
- gpu