diff --git a/docker/dockerfile.torch b/docker/dockerfile.torch index 086b62bfb..6866fb594 100644 --- a/docker/dockerfile.torch +++ b/docker/dockerfile.torch @@ -33,5 +33,7 @@ COPY --chown=1000:1000 --from=dlfw /opt/conda/lib/python3.8/site-packages/numba COPY --chown=1000:1000 --from=dlfw /opt/conda/lib/python3.8/site-packages/numpy /usr/local/lib/python3.8/dist-packages/numpy COPY --chown=1000:1000 --from=dlfw /opt/conda/lib/python3.8/site-packages/torch /usr/local/lib/python3.8/dist-packages/torch +RUN pip install --no-deps fastai fastcore fastprogress fastdownload + HEALTHCHECK NONE CMD ["/bin/bash"] \ No newline at end of file