fabiwlf / torch-m1

PyTorch for ARM64 (works on M1 Macs) / x64 and raspberry Pi etc.

📈 PyTorch for M1 Macs and ARM64 Devices

PyTorch for ARM64 (works on M1 Macs) / x64 and raspberry Pi etc.

Usage

Using Image from GitHub packages

docker pull https://github.com/fabiwlf/torch-m1

Running example

# 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

Building Image from source

docker buildx build -f Dockerfile -t https://github.com/fabiwlf/torch-m1 --load # from Dockerfile