content/piper/src/python/Dockerfile.txt

FROM nvcr.io/nvidia/pytorch:22.03-py3

RUN pip3 install \
    'pytorch-lightning~=1.7.0'

ENV NUMBA_CACHE_DIR=.numba_cache