FROM nvcr.io/nvidia/pytorch:22.03-py3 RUN pip3 install \ 'pytorch-lightning~=1.7.0' ENV NUMBA_CACHE_DIR=.numba_cache