Upgrade Python and Pytorch version in container image for pytorch-mnist example

[tenzen-y]

/kind feature

Describe the solution you'd like
[A clear and concise description of what you want to happen.]
As discussed in #1882 (comment), we need to upgrade Python and Pytorch version.

/cc @johnugeorge

Anything else you would like to add:
[Miscellaneous information that will assist in solving the issue.]

---

Love this feature? Give it a 👍 We prioritize the features with the most 👍

[tenzen-y]

Also, we need to upgrade the python and Pytorch versions for darts-cnn-cifar10.

/assign