Skip to content

Conversation

@edoardob90
Copy link
Member

  • Reduce image size by implementing multi-stage build
  • Remove torchaudio package and optimize PyTorch installation
  • Add GPU/CPU variant support via build arguments
  • Enable multi-architecture builds (amd64/arm64) with proper manifests
  • Fix Renku compatibility by setting provenance: false
  • Simplify container environment variables and working directory
  • Create architecture-specific tags for better image management

edoardob90 added 9 commits May 6, 2025 11:03
- Reduce image size by implementing multi-stage build
- Remove torchaudio package and optimize PyTorch installation
- Add GPU/CPU variant support via build arguments
- Enable multi-architecture builds (amd64/arm64) with proper manifests
- Fix Renku compatibility by setting provenance: false
- Simplify container environment variables and working directory
- Create architecture-specific tags for better image management
* Fix syntax
* Force PyTorch uninstall first
@edoardob90 edoardob90 marked this pull request as ready for review May 6, 2025 12:07
yakutovicha
yakutovicha previously approved these changes May 6, 2025
Copy link
Member

@yakutovicha yakutovicha left a comment

Choose a reason for hiding this comment

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

LGTM!

In Docker meta step "prefix=" was empty
@edoardob90
Copy link
Member Author

Maybe we should also adjust the README? Saying that latest is now pulling the CPU image by default. If you want the other variant, should specify cuda

@edoardob90 edoardob90 merged commit 176e30b into main May 6, 2025
1 check passed
@edoardob90 edoardob90 deleted the optimize-multiarch-docker-images branch May 6, 2025 12:55
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.

3 participants