lightningdot
copied
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
23 lines
646 B
23 lines
646 B
3 years ago
|
FROM nvcr.io/nvidia/pytorch:19.05-py3
|
||
|
|
COPY requirements.txt scripts/download_bert.py ./
|
||
|
|
RUN pip install -r requirements.txt &&\
|
||
|
|
python download_bert.py &&\
|
||
|
|
rm ./requirements.txt ./download_bert.py
|
||
|
|
|
||
|
|
################## v1 ##########################
|
||
|
|
|
||
|
|
COPY scripts/install_horovod.sh ./
|
||
|
|
RUN source install_horovod.sh &&\
|
||
|
|
rm ./install_horovod.sh
|
||
|
|
ENV OPENMPI_VERSION=4.0.0
|
||
|
|
|
||
|
|
# fix ssh permissions
|
||
|
|
RUN bash -c "chmod -R 600 /etc/ssh/ && chmod 600 /var/run/sshd/ && chmod 600 /root"
|
||
|
|
|
||
|
|
################## horovod, v2 ##########################
|
||
|
|
|
||
|
|
|
||
|
|
RUN bash -c "pip install lz4==2.1.9 lmdb==0.97"
|
||
|
|
|
||
|
|
################# LMDB ##########################
|