7 lines
123 B
Docker
7 lines
123 B
Docker
|
FROM nvcr.io/nvidia/pytorch:22.03-py3
|
||
|
|
|
||
|
|
RUN pip3 install \
|
||
|
|
'pytorch-lightning~=1.7.0'
|
||
|
|
|
||
|
|
ENV NUMBA_CACHE_DIR=.numba_cache
|