version: 0.1 entrypoint: python runway_model.py python: 3.6 cuda: 10.2 framework: tensorflow spec: gpu: True cpu: False files: ignore: - checkpoints/* build_steps: - apt-get update - apt-get upgrade -y > /dev/null - apt-get install -y cuda-nvcc-10-2 cuda-toolkit-10-2 - pip install -r requirements.txt