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

feat(docker): install CUDA drivers in autoware:base image #5159

Merged
merged 2 commits into from
Sep 4, 2024

Conversation

youtalk
Copy link
Member

@youtalk youtalk commented Sep 2, 2024

Description

Based on the feedback from https://github.com/orgs/autowarefoundation/discussions/5110#discussioncomment-10469371, this PR will create an autoware:base-cuda image that installs the CUDA drivers on the base stage. By installing several gigabytes of CUDA drivers in the top stage of the Dockerfile, the build cache will be utilized more effectively, making the docker pull cache easier to leverage.

Tests performed

https://github.com/autowarefoundation/autoware/actions/runs/10667352309

Effects on system behavior

Not applicable.

Interface changes

Pre-review checklist for the PR author

The PR author must check the checkboxes below when creating the PR.

In-review checklist for the PR reviewers

The PR reviewers must check the checkboxes below before approval.

Post-review checklist for the PR author

The PR author must check the checkboxes below before merging.

  • There are no open discussions or they are tracked via tickets.

After all checkboxes are checked, anyone who has write access can merge the PR.

Signed-off-by: Yutaka Kondo <yutaka.kondo@youtalk.jp>
Signed-off-by: Yutaka Kondo <yutaka.kondo@youtalk.jp>
@youtalk youtalk self-assigned this Sep 2, 2024
@youtalk youtalk added type:containers Docker containers, containerization of components, or container orchestration. component:openadkit Issues or Features related to Open AD Kit tag:run-health-check Run health-check labels Sep 2, 2024
@youtalk youtalk marked this pull request as ready for review September 2, 2024 12:43
@youtalk
Copy link
Member Author

youtalk commented Sep 3, 2024

With this change, the image sizes of autoware-core-cuda and autoware-universe-cuda will be reduced by 3GB.

Screenshot from 2024-09-03 15-02-47

Copy link
Member

@mitsudome-r mitsudome-r left a comment

Choose a reason for hiding this comment

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

LGTM

@youtalk youtalk merged commit d5568e0 into main Sep 4, 2024
34 checks passed
@youtalk youtalk deleted the upstream-base-cuda branch September 4, 2024 01:56
ENV CCACHE_DIR="/root/.ccache"

# Set up development environment
RUN --mount=type=ssh \
--mount=type=cache,target=/var/cache/apt,sharing=locked \
./setup-dev-env.sh -y --module all ${SETUP_ARGS} --no-cuda-drivers openadkit \
./setup-dev-env.sh -y --module all openadkit \
Copy link

Choose a reason for hiding this comment

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

@youtalk this command will not run the autoware.dev_env.cuda role.
The autoware.dev_env.cuda will only run if module == 'base' and prompt_install_nvidia=='y'.
Is it possible that there is a bug here?

Copy link
Member Author

Choose a reason for hiding this comment

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

Copy link

Choose a reason for hiding this comment

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

@youtalk what is misleading is the argument --no-cuda-drivers which implies that there are no CUDA drivers?.

Copy link
Member Author

Choose a reason for hiding this comment

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

youtalk added a commit that referenced this pull request Oct 11, 2024
Signed-off-by: Yutaka Kondo <yutaka.kondo@youtalk.jp>
youtalk added a commit to youtalk/autoware that referenced this pull request Oct 11, 2024
Signed-off-by: Yutaka Kondo <yutaka.kondo@youtalk.jp>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
component:openadkit Issues or Features related to Open AD Kit tag:run-health-check Run health-check type:containers Docker containers, containerization of components, or container orchestration.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants