Commit 852ad811 authored by blackheaven's avatar blackheaven
Browse files

Update docker-compose.yml

parent a90a016c
Loading
Loading
Loading
Loading

docker-compose.yml

0 → 100644
+24 −0
Original line number Diff line number Diff line
# https://pallawi-ds.medium.com/tensorflow-gpu-jupyterlab-docker-container-how-to-access-files-using-jupyter-lab-on-localhost-ba4ff281e4a4
# https://kyumdoctor.co.kr/18
version: '3.3'
services:
  jupyter-gpu:
    container_name: jupyter-gpu
    ports:
      - '29999:9999'
    volumes:
      - '${PWD}/tf:/tf'
      - '${PWD}/home:/home'
    image: 'tensorflow/tensorflow:latest-gpu'
    # user: 1000:1000
    command: tail -f /dev/null
    restart: always
    # command: jupyterhub --ip 0.0.0.0 --port 9999
    # runtime: nvidia
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]