Skip to content

[Notice] Removing "torch" package from the "universal" image #856

@samruddhikhandale

Description

@samruddhikhandale

Hello 👋

We are thinking of removing the torch package from the universal image. Feel free to share your concerns or thoughts, thank you! ✨

Background

Currently, the size of universal image is 12.7 GB and we have been looking at ways to shrink it. It’s interesting to see that the machine learning packages consume the most space which is 5.24 GB. We understand, that we want to support ML relevant scenarios, however, 41% kinda seems like a big chunk of the image.

Alternative

As torch package is installed with pip, it is very simple to add to your dev container (either in a Dockerfile or with a lifecycle command).

Also, we are thinking of adding a new Feature option to the Python Feature (eg. packages) which accepts a list of packages which can be added with pip. The user can install torch or any other packages they would like to with the update Python Feature. Would appreciate any feedbacks on this as well.

Conclusion

The torch package is 1.4 GB and it’s add a dependency of nvidia which is another 3.0 GB . Hence, removing the torch package saves us ~ 4.4 GB.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions