/docker-pytorch-playground

Running models faster and easier, support for x86 and ARM (M1, M1Pro).

Primary LanguageDockerfileApache License 2.0Apache-2.0

Docker Pytorch Playground

Running models faster and easier, support for x86 and ARM (M1, M1Pro).

How to Use

It's easy to use:

docker pull soulteary/docker-pytorch-playground

Docker Images

  • soulteary/docker-pytorch-playground:latest
  • soulteary/docker-pytorch-playground:2022.05.19

Build

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 .