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

Use Colab as a base image. #1444

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Use Colab as a base image. #1444

wants to merge 3 commits into from

Conversation

djherbis
Copy link
Contributor

This change makes a number of major changes:

  • Colab is the base image
  • uv is the main package install tool
  • leveraging requirements.txt instead of many separate installs
  • stop building and installing tensorflow/torch/lightbgm/jax since those are managed by the Colab base image now

In order to decide what packages to explicitly install I:

  • looked at what packages are in the Colab base image
  • looked at what packages were in the Kaggle image
  • looked at what packages were explicitly mentioned in Kaggle Dockerfile

This may still take a few iterations to get all the right parts in the image, but this should hopefully make the image much more manageable.

http://b/365782129

This change makes a number of major changes:
- Colab is the base image
- uv is the main package install tool
- leveraging requirements.txt instead of many separate installs
- stop building and installing tensorflow/torch/lightbgm/jax since those
  are managed by the Colab base image now

In order to decide what packages to explicitly install I:
- looked at what packages are in the Colab base image
- looked at what packages were in the Kaggle image
- looked at what packages were explicitly mentioned in Kaggle Dockerfile

This may still take a few iterations to get all the right parts in the
image, but this should hopefully make the image much more manageable.

http://b/365782129
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.

1 participant