PyTorch ARM64 Size

1

I was trying to use PyTorch on AWS Lambda and I have an interesting observation whereby the torch package on a x86_64 OS docker image is 1.7G in size while being only 207MB in size on a arm64 OS docker image.

What would cause such a huge difference in file fize?

2

I’m not sure which containers you are comparing but I cannot find an ARM64 container in docker hub.
Based on the size difference I would guess the x86 container ships with a CUDA toolkit (and other additional libs).

3

Oh I am using the one here Docker

4

Ah OK. I don’t know how these images are built.