Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Modify other Dockerfiles with 2stage and conda-pack improvements #148

Merged
merged 1 commit into from
Apr 11, 2022

Conversation

af-a
Copy link
Contributor

@af-a af-a commented Apr 11, 2022

Description

This is an extension of #145, where the Dockerfile_reconsurf was improved with multi-stage builds, integrating conda-pack, and installing packages using yaml files.

Here, the other Dockerfiles are modified to include the same changes.
The sizes of images produced by each Dockerfile are now as follows:

  • Dockerfile: 12.1GB
  • Dockerfile_CPU: 4.96GB
  • Dockerfile_FastSurferCNN: 11.5GB
  • Dockerfile_FastSurferCNN_CPU: 4.42GB
  • Dockefile_reconsurf: 1.56GB

Follow-up Items

(1) The base image for the GPU-supporting images may be switched from nvidia/cuda:11.1-cudnn8-runtime-ubuntu20.04 to ubuntu20.04 soon, since a global CUDA+CuDNN installation may be unnecessary for pytorch installed with cudatoolkit. With this change, these images will shrink considerably, and all images will share the same base image.
(2) Once (1) is included, the Dockerfiles are quite similar except for the input yaml files and whether FreeSurfer is installed or not. They could be reduced to a single file with docker build arguments, such as WITH_CUDA and WITH_FS.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants