version: "3.9"

services:
  pytorch:
    build: .
    container_name: pytorch-lab
    ports:
      - "8888:8888"  # Jupyter
      - "6006:6006"  # TensorBoard
      - "8097:8097"  # Visdom
    volumes:
      - .:/workspace
    working_dir: /workspace
    command: jupyter lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root