Update python.md

[gregordecristoforo]

The link to the ROCM pytorch container is removed as it points to the latest ROCM container which is incompatible with LUMI's ROCM drivers.

[mszpindler]

So this is almost exactly the same piece of text as in https://github.com/Lumi-supercomputer/LUMI-AI-Guide/blob/3c45859fc22054095a1679351cc0a9289695d0df/2-setting-up-environment/README.md#containers-on-lumi

[hdrei]

Reminding @gregordecristoforo and @mszpindler about this open PR

[mszpindler]

I would like to simplify the page and moving container details away:

• link to base images list instead of describing here
• link to details on how to extend existing base images
• verify if I do not compacted cotainr and the wrapper description too much