/onnx-docker

Dockerfiles and scripts for ONNX container images

Primary LanguageJupyter NotebookMIT LicenseMIT

What is this repository for?

To store the docker BUILD scripts of ONNX related docker images.

  • onnx-base: Use published ONNX package from PyPi with minimal dependencies.
  • onnx-dev: Build ONNX from source with minimal dependencies.
  • onnx-ecosystem: Jupyter notebook environment for getting started quickly with ONNX models, ONNX converters, and inference using ONNX Runtime.

Docker Image Workflow

  1. Obtain the Docker images

You can clone this repository and build your desired image.

# onnx-base container
cd onnx-base
docker build . -t onnx-base

# onnx-developer container
cd onnx-dev
docker build . -t onnx-dev

# onnx-ecosystem container
cd onnx-ecosystem
docker build . -t onnx-ecosystem

Alternatively, you can pull a pre-built image from DockerHub.

docker pull onnx/onnx-base
docker pull onnx/onnx-dev
docker pull onnx/onnx-ecosystem
  1. Run the images
docker images

docker run -it onnx-base
docker run -it onnx-dev
docker run -p 8888:8888 onnx-ecosystem