Skip to content

Latest commit

 

History

History
39 lines (28 loc) · 1.08 KB

README.md

File metadata and controls

39 lines (28 loc) · 1.08 KB

What is this repository for?

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

  • onnx-docker-cpu: Image with ONNX, PyTorch, Tensorflow support
  • onnx-docker-gpu: Image with ONNX, PyTorch (CUDA), Caffe2 support
  • 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-docker-cpu container
docker build onnx-docker/onnx-docker-cpu

# onnx-docker-gpu container
docker build onnx-docker/onnx-docker-gpu

# 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-docker:branch-name

docker pull onnx/onnx-ecosystem
  1. Run the images
docker images
docker run -i -t image_id

docker run -p 8888:8888 onnx-ecosystem