Investigate issue with specific Pytorch versions on Mac M1

[C-Achard]

An issue similar to the one described here occurred when testing on an M1 Mac.

This will need to be investigated so we can set the appropriate versions of torch in the ARM64 env.