PyTorch for ARM64 (works on M1 Macs) / x64 and raspberry Pi etc.
docker pull https://github.com/fabiwlf/torch-m1
# This will start a container and mount the "example" folder to it
docker run -it --rm -v $(pwd)/example:/app https://github.com/fabiwlf/torch-m1
docker buildx build -f Dockerfile -t https://github.com/fabiwlf/torch-m1 --load # from Dockerfile