Windows ML Dev Env /w Docker

compose.gpu.yml

services:
  psj:
    image: nvcr.io/nvidia/pytorch:24.06-py3
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
              driver: nvidia  
    shm_size: 10gb
    ulimits:
      memlock: -1
    stdin_open: true
    tty: true
    volumes:
      - .:/home/psj