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

Update GPU dockerfile to fully support GPU workflow #26

Merged
merged 28 commits into from
May 21, 2024
Merged

Conversation

forrestfwilliams
Copy link
Contributor

@forrestfwilliams forrestfwilliams commented May 17, 2024

  • Fixed Dockerfile.gpu so that outputs will contain actual data.
  • Refactored scripts/build_proc.sh to combine GPU compilation steps.
  • Final product zip archive is now always created.
  • Added scripts/ubuntu_setup.sh for setting up a GPU-based Ubuntu EC2 AMI.
  • Updated README.md to reflect the above changes.

@forrestfwilliams forrestfwilliams requested a review from a team as a code owner May 17, 2024 15:23
@forrestfwilliams forrestfwilliams changed the title Update GPU dockerfile to support correct compute capability Update GPU dockerfile to fully support GPU workflow May 20, 2024
ENV DEBIAN_FRONTEND=noninteractive

# FIXME: can remove git after switch back to released version of back-projection
RUN apt-get update && apt-get install -y --no-install-recommends unzip vim curl git build-essential gfortran libfftw3-dev && \
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Do you need vim?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is standard for our plugins (see the cookiecutter Dockerfile). It's a pretty minimal package but provides of a lot of benefits when debugging within the container.


# FIXME: can remove git after switch back to released version of back-projection
RUN apt-get update && apt-get install -y --no-install-recommends unzip vim curl git build-essential gfortran libfftw3-dev && \
apt-get clean && rm -rf /var/lib/apt/lists/*
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why rm -rf /var/lib/apt/lists/*?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This removes unneeded apt metadata that serves no purpose other than bloating the container.

Comment on lines +3 to +4
# MULTIARCH_DIR=/usr/lib/$(gcc -print-multiarch)
# FFTW_LIB=$MULTIARCH_DIR/libfftw3f.a
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Are the commented-out lines still needed?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yeah, we may need to revert to this at some point, so I'd rather keep these close at hand.

scripts/build_proc.sh Outdated Show resolved Hide resolved
Copy link
Contributor

@jtherrmann jtherrmann left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, feel free to merge unless you think any of my comments need to be addressed.

@forrestfwilliams forrestfwilliams merged commit c87f10a into develop May 21, 2024
9 checks passed
@forrestfwilliams forrestfwilliams deleted the g6_gpu branch May 21, 2024 17:17
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