Skip to content

Latest commit

 

History

History
12 lines (9 loc) · 442 Bytes

pytorch_caffe2_docker.md

File metadata and controls

12 lines (9 loc) · 442 Bytes
Raw

Caffe2/PyTorch Docker

Docker images (CPU-only and GPU versions) with ONNX, PyTorch, and Caffe2 are available for quickly trying tutorials that use ONNX. To quickly try CPU-only version, simply run:

docker run -it --rm onnx/onnx-docker:cpu /bin/bash

To run the version with GPU support, nvidia-docker is needed. Execute:

nvidia-docker run -it --rm onnx/onnx-docker:gpu /bin/bash