| FROM zeahub/all:v0.0.4 | |
| RUN apt-get update && apt-get install -y git-lfs && rm -rf /var/lib/apt/lists/* | |
| RUN pip install --no-cache-dir gradio tyro optuna | |
| RUN pip install --no-cache-dir --no-deps pytorch_fid | |
| RUN pip install --no-cache-dir -U keras | |
| WORKDIR /workspace | |