build_gpu.yml

version: '2.3'

services:

  docker_sdk:
    image: tf2_docker_sdk_api
    build:
      context: ./docker_sdk_api
      dockerfile: docker/Dockerfile
      args:
        http_proxy: 
        https_proxy:
  
  training_api:
    image: tf2_training_api_gpu
    build:
      context: ./training_api
      dockerfile: docker/GPU/Dockerfile
      args:
        http_proxy:
        https_proxy:
  
  user_interface:
    image: tf2_gui
    build:
      context: ./gui
      dockerfile: ./Dockerfile
      args:
        http_proxy:
        https_proxy:

  inference_api:
    image: tf2_inference_api_cpu
    build:
      context: ./inference_api
      dockerfile: docker/dockerfile
      args:
        http_proxy:
        https_proxy: