Skip to content

Instantly share code, notes, and snippets.

@ovuruska

ovuruska/Dockerfile

Created February 4, 2021 08:38
Show Gist options
  • Save ovuruska/f3854e575e901c24de0745f143770347 to your computer and use it in GitHub Desktop.
Save ovuruska/f3854e575e901c24de0745f143770347 to your computer and use it in GitHub Desktop.
Download ZIP
Run Torchtrt with custom volume
Raw
Dockerfile
FROM nvcr.io/nvidia/pytorch:20.03-py3
RUN git clone https://github.com/NVIDIA-AI-IOT/torch2trt
RUN cd torch2trt && python setup.py install --plugins
VOLUME ./jetson/ ./jetson/
WORKDIR /workspace
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment