Fix up docker

This commit is contained in:
Michael Hansen
2023-02-24 12:45:57 -06:00
parent 131dbf4b5b
commit 5493e5dd4c
2 changed files with 7 additions and 7 deletions

View File

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