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

Build with requirements.txt #43

Closed
wants to merge 9 commits into from
Closed

Conversation

Veldhoen
Copy link
Member

Image build failed frequently, because of heavy dependencies, mostly: Torch.
So now, I opted to use a Pytorch base image.

This works fine with a separate equirements.txt file and pip installation.

However, when I generate the requirements.txt from poetry automatically, it will create dependency on Cuda 12 instead of 11. I can't figure out how to make poetry respect the fact that torch and Cuda support are already available. Is there a way to keep in sync with the pyproject.toml/poetry.lock?

@Veldhoen Veldhoen linked an issue Jul 22, 2024 that may be closed by this pull request
4 tasks
@Veldhoen Veldhoen requested a review from gb-beng July 22, 2024 11:41
@gb-beng
Copy link
Contributor

gb-beng commented Jul 23, 2024

Tried our multi-stage build approach, and it works well! #44. I'd go for that, since its a lot simpler.

@gb-beng gb-beng closed this Jul 23, 2024
@gb-beng gb-beng mentioned this pull request Jul 23, 2024
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.

Use Torch base image
2 participants