99f9967
1
2
3
4
5
6
7
FROM nvcr.io/nvidia/pytorch:22.03-py3 RUN pip3 install \ 'pytorch-lightning~=1.7.0' ENV NUMBA_CACHE_DIR=.numba_cache
