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
@@ -35,7 +35,7 @@ RUN wget https://mirrors.huaweicloud.com/python/3.9.7/Python-3.9.7.tgz && \
 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.tuna.tsinghua.edu.cn/simple && pip install numpy==1.22.4 -i https://pypi.tuna.tsinghua.edu.cn/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,10 +50,10 @@ 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 install -y 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

@@ -18,16 +18,23 @@
     # 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 构建镜像"
 
-    如果你对 docker 有一定了解，则可以通过我们提供的 Dockerfile 文件，直接构建出能运行 PaddleScience 的环境。按照下列步骤构建镜像，创建容器并进入容器，以运行 PaddleScience。
+    ``` 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
+    ```
 
-    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. **在容器内手动执行：`{++ldconfig++}`，刷新动态链接库缓存，否则可能出现 GPU 设置不正确的问题。**
+    如果出现因网络问题导致的 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