version: '3.9'
services:
  jupyter-tensorflow:
    image: tensorflow/tensorflow:latest-gpu-jupyter
    container_name: jupyter-tensorflow
    restart: always
    volumes:
      - jupyter-folder:/tf
    ports:
      - "127.0.0.1:8888:8888"
    environment:
      - HOME=/tf
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

volumes:
  jupyter-folder:
    driver: local
    driver_opts:
      o: bind
      type: none
      device: /home/margui/docker/JUPYTER-TENSORFLOW/data