Running models faster and easier, support for x86 and ARM (M1, M1Pro).
It's easy to use:
docker pull raykr/docker-pytorch-playground
- raykr/docker-pytorch-playground:latest
docker build -t raykr/docker-pytorch-playground .
If you need to replace the software source:
docker build --build-arg USE_MIRROR=true -t raykr/docker-pytorch-playground .