mirror of
https://github.com/pstrueb/piper.git
synced 2026-04-17 22:05:30 +00:00
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
|