docker: Allow caching packages for other arches

When running on the CI, we want to be able to cache packages for arches
other than the arch we are running on, so allow optional --arch to be
passed to the script.

Signed-off-by: Ryan Roberts <ryan.roberts@arm.com>