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