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