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.
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 GBand we have been looking at ways to shrink it. It’s interesting to see that the machine learning packages consume the most space which is5.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
torchpackage is installed withpip, 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 withpip. The user can installtorchor any other packages they would like to with the update Python Feature. Would appreciate any feedbacks on this as well.Conclusion
The
torchpackage is1.4 GBand it’s add a dependency ofnvidiawhich is another3.0 GB. Hence, removing thetorchpackage saves us~ 4.4 GB.