[bug] docker images are only published for arm

### Enter the chapter number

N/A

### Enter the page number

_No response_

### What is the cell's number in the notebook

_No response_

### Enter the environment you are using to run the notebook

Other (specify in description)

### Describe your issue

Running the following command should have worked on my amd64 workstation:

```shell
$ docker run -it --rm --gpus all --name homlp -p 8888:8888 -v "$PWD":/content ageron/handson-mlp:cu126
WARNING: The requested image's platform (linux/arm64/v8) does not match the detected host platform (linux/amd64/v4) and no specific platform was requested
exec /opt/venvs/handson-mlp/bin/jupyter: exec format error
```

Docker images are not published for amd64:

### Enter what you expected to happen

The images published on Docker Hub to be multi-arch.

### If you found a workaround, describe it here

Rebuilt an image myself, albeit re-wrote it with these changes that could be detailed in an enhancement issue which I can submit PR(s) if desired, but is there a way to test the image other than manually running all the notebooks?

- Multi-stage building
- Docker mount caches for apt/uv
- Non-root
- Add notebook-dir param for jupyter
- Updated pytorch dep to use 12.8 driver and `nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04` base image