hariharans29 / onnx-docker

Dockerfiles and scripts for ONNX container images

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

2. Run the images

docker images
docker run -i -t image_id

docker run -p 8888:8888 onnx-ecosystem