FROM gcr.io/deeplearning-platform-release/tf2-cpu.2-2:latest

RUN pip install --user --no-cache-dir boto3

RUN pip install --user --no-cache-dir torch-tb-profiler