ci: update Dockerfile

This commit is contained in:
zweien 2021-07-12 15:40:33 +08:00
parent 84cbcce755
commit 8f92888033
1 changed files with 6 additions and 26 deletions

View File

@ -1,27 +1,7 @@
FROM pytorch/pytorch:1.7.0-cuda11.0-cudnn8-devel
RUN apt-get update && apt-get install -y openssh-server nfs-common && \
echo "PermitRootLogin yes" >> /etc/ssh/sshd_config && \
(echo '123456'; echo '123456') | passwd root
RUN pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ transforms3d \
typing \
numpy \
keras \
h5py \
pandas \
zipfile36 \
scikit-optimize \
pytest \
sphinx \
matplotlib \
myst_parser \
sphinx_rtd_theme==0.5.2 \
tensorboard==2.4.1 \
sympy==1.5.1 \
pyevtk==1.1.1 \
flask==1.1.2 \
requests==2.25.0 \
networkx==2.5.1
COPY . /idrlnet/
RUN cd /idrlnet && pip install -e .
ENTRYPOINT service ssh start && bash
LABEL maintainer="pengwei"
WORKDIR /idrlnet
COPY requirements.txt ./
RUN pip install -r requirements.txt
COPY . .
RUN pip install -e .