[Docker] Add docker image (#856)

* execute 'ldconfig' after /bin/bash automatically

* add_dockerhub

* refine doc

* update base image to paddle official image(test=document_fix)

Author: HydrogenSulfate

docker/Dockerfile

@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/cuda:11.6.2-cudnn8-runtime-ubuntu20.04
+FROM paddlepaddle/paddle:latest-dev-cuda11.6-cudnn8.4-trt8.4-gcc82
 
 RUN echo "deb http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse" >> /etc/apt/sources.list
 RUN echo "deb-src http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse" >> /etc/apt/sources.list
@@ -32,10 +32,10 @@ RUN wget https://mirrors.huaweicloud.com/python/3.9.7/Python-3.9.7.tgz && \
     cd .. && rm -rf Python-3.9.7 && rm Python-3.9.7.tgz && \
     ln -s /usr/local/bin/python3.9 /usr/local/bin/python
 
-RUN python3 -m pip install --upgrade pip -i https://pypi.douban.com/simple/ && pip install pysdf open3d db-sqlite3 -i https://pypi.douban.com/simple/
+RUN python3 -m pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple && pip install pysdf open3d db-sqlite3 -i https://pypi.tuna.tsinghua.edu.cn/simple
 
 # install pymesh and paddle
-RUN python3 -m pip install --upgrade pip -i https://pypi.douban.com/simple/ && pip install numpy==1.22.4 -i https://pypi.douban.com/simple/
+RUN python3 -m pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple && pip install "numpy>=1.20.0,<=1.23.1" -i https://pypi.tuna.tsinghua.edu.cn/simple
 COPY ./pymesh.tar.xz /src/pymesh.tar.xz
 
 RUN cd /src && \
@@ -50,8 +50,11 @@ ENV PATH="$PATH:/usr/lib/x86_64-linux-gnu/"
 ENV LD_LIBRARY_PATH="/usr/local/cuda/lib64/:$LD_LIBRARY_PATH"
 
 RUN python3 -m ensurepip --upgrade && \
-    pip3 config set global.index-url https://pypi.doubanio.com/simple/ && \
+    pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
     # ensure paddle install normally
-    ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.8 /usr/local/cuda/lib64/libcudnn.so && \
-    ln -s /usr/local/cuda/lib64/libcublas.so.11 /usr/local/cuda/lib64/libcublas.so && \
     python3 -m pip install paddlepaddle-gpu==0.0.0.post116 -f https://www.paddlepaddle.org.cn/whl/linux/gpu/develop.html
+
+RUN apt-get update
+RUN apt-get install -y -o Acquire::Retries=10 --no-install-recommends openssh-server
+
+RUN echo 'ldconfig' >> /root/.bashrc

docker/run.sh

@@ -1,9 +1,9 @@
 docker build . -t paddlescience:latest
 
 if [ -x "$(command -v nvidia-docker)" ]; then
-    nvidia-docker run --network=host -it paddlescience
+    nvidia-docker run --name paddlescience_container --network=host -it paddlescience
 elif [ -x "$(command -v docker)" ]; then
-    docker run --gpus all --network=host -it paddlescience
+    docker run --name paddlescience_container --gpus all --network=host -it paddlescience
 else
     echo "Docker start failed, please install nvidia-docker or docker(>=19.03) first"
 fi

docs/zh/install_setup.md

@@ -4,14 +4,37 @@
 
 ### 1.1 从 docker 镜像启动[可选]
 
-如果你对 docker 有一定了解，则可以通过我们提供的 Dockerfile 文件，直接构建出能运行 PaddleScience 的环境。按照下列步骤构建镜像，创建容器并进入容器，以运行 PaddleScience。
+=== "从 DockerHub 拉取镜像"
 
-1. 克隆 PaddleScience 源码：`git clone https://github.com/PaddlePaddle/PaddleScience.git`
-2. 下载 PyMesh 预编译文件压缩包 [pymesh.tar.xz](https://paddle-org.bj.bcebos.com/paddlescience/docker/pymesh.tar.xz)，并放置在 `PaddleScience/docker/` 目录下
-3. 在 `PaddleScience/docker/` 目录下，执行 `bash run.sh`，等待 docker build 完毕后自动进入环境。如果出现因网络问题导致的 apt 下载报错，则重复执行 `bash run.sh` 直至 build 完成即可
-4. 在 docker 环境中，执行 `ldconfig`
+    ``` sh
+    # pull image
+    docker pull hydrogensulfate/paddlescience
+
+    # create a container named 'paddlescience' based on pulled image
+    ## docker version < 19.03
+    nvidia-docker run --name paddlescience_container --network=host -it --shm-size 64g hydrogensulfate/paddlescience:latest /bin/bash
+
+    ## docker version >= 19.03
+    # docker run --name paddlescience_container --gpus all --network=host -it shm-size 64g hydrogensulfate/paddlescience:latest /bin/bash
+    ```
+
+    !!! note
+
+        Dockerhub 拉取的镜像**仅**预装了运行 PaddleScience 所需的依赖包，如 pymesh、open3d，**并不包含 PaddleScience**。
+        因此请在镜像拉取和容器构建完成后，参考 [1.4 安装 PaddleScience](#14-paddlescience) 中的步骤，在容器中安装 PaddleScience。
+
+=== "通过 Dockerfile 构建镜像"
+
+    ``` sh
+    git clone https://github.com/PaddlePaddle/PaddleScience.git
+    cd PaddleScience/docker/
+    wget -nc https://paddle-org.bj.bcebos.com/paddlescience/docker/pymesh.tar.xz
+    bash run.sh
+    ```
+
+    如果出现因网络问题导致的 docker 构建时 apt 下载报错，则重复执行 `bash run.sh` 直至构建完成。
 
-更多关于 Paddle Docker 的安装和使用，请参考 [Docker 安装](https://www.paddlepaddle.org.cn/documentation/docs/zh/install/docker/fromdocker.html)
+    更多关于 Paddle Docker 的安装和使用，请参考 [Docker 安装](https://www.paddlepaddle.org.cn/documentation/docs/zh/install/docker/fromdocker.html)。
 
 ### 1.2 python 环境安装[可选]
 

examples/NLS-MB/conf/NLS-MB_rogue_wave.yaml

@@ -60,7 +60,7 @@ EVAL:
   eval_with_no_grad: true
 
 INFER:
-  pretrained_model_path: null
+  pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/NLS-MB/NLS-MB_rogue_wave_pretrained.pdparams
   export_path: ./inference/NLS-MB_rogue_wave
   pdmodel_path: ${INFER.export_path}.pdmodel
   pdpiparams_path: ${INFER.export_path}.pdiparams

examples/NLS-MB/conf/NLS-MB_soliton.yaml

@@ -59,7 +59,7 @@ EVAL:
   eval_with_no_grad: true
 
 INFER:
-  pretrained_model_path: null
+  pretrained_model_path: https://paddle-org.bj.bcebos.com/paddlescience/models/NLS-MB/NLS-MB_soliton_pretrained.pdparams
   export_path: ./inference/NLS-MB_soliton
   pdmodel_path: ${INFER.export_path}.pdmodel
   pdpiparams_path: ${INFER.export_path}.pdiparams