Skip to content

Latest commit

 

History

History
23 lines (18 loc) · 1006 Bytes

INSTALL.md

File metadata and controls

23 lines (18 loc) · 1006 Bytes
Raw

Use Docker image

We have set up a docker image: fotstrt/orion-ae with all packages pre-installed. We assume NVIDIA drivers are installed in the source machine, and that docker containers can use the host machine's GPUs.

  • Start a container with docker run --gpus=1 -it fotstrt/orion-ae:v1 bash
  • Download the Orion repo and install:
    • git clone https://github.com/eth-easl/orion.git
    • cd orion
    • bash compile.sh
    • pip install -e .

Without Docker image

In order to use Orion without our pre-built image, a user must install:

  • NVIDIA CUDA. We have tested Orion with CUDA 10.2 and CUDA 11.3
  • (optionally) NVIDIA CUDNN
  • Pytorch (from source) + TorchVision
  • Download the Orion repo and install:
    • git clone https://github.com/eth-easl/orion.git
    • cd orion
    • bash compile.sh
    • pip install -e .