Skip to content

Dockerfiles and scripts for ONNX container images

License

Notifications You must be signed in to change notification settings

hariharans29/onnx-docker

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

44 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Dockerfiles and scripts for ONNX container images

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 90.0%
  • Dockerfile 10.0%