ROCm-docker icon indicating copy to clipboard operation
ROCm-docker copied to clipboard

ROCm pytorch images size

Open 88Ocelot opened this issue 1 year ago • 1 comments

The current size of the ROCm PyTorch Docker image, particularly rocm/pytorch:latest, is significantly large, consuming approximately 54 GB on disk when uncompressed. A more streamlined version, rocm-torch-slim, has been created with a reduced size of around 28 GB through the use of a multistage build(Dockerfile). Further optimizations are possible, and experimenting with the configuration may lead to an even smaller image size.

BTW: Why do we need to install ROCm binaries on the host system if the Docker image performs the heavy lifting?

88Ocelot avatar Dec 25 '23 12:12 88Ocelot