nv-morpheus · Jul 15, 2022 · Jun 22, 2022 · Jun 23, 2022 · Jun 23, 2022 · Jun 23, 2022
@@ -44,15 +44,13 @@ pipeline {
                         timeout(time: 4, unit: 'HOURS')
                     }
                     environment {
-                        PARALLEL_LEVEL = '10'
                         USE_SCCACHE = '1'
                         HOME = "${WORKSPACE}"
                     }
                     agent {
                         docker {
-                            image 'gpuci/rapidsai:22.04-cuda11.5-devel-centos7-py3.8'
-                            label 'driver-495'
-                            args '--runtime "nvidia" -e "NVIDIA_VISIBLE_DEVICES=$EXECUTOR_NUMBER"'
+                            image 'gpuci/rapidsai-driver:22.04-cuda11.5-devel-centos7-py3.8'
+                            label 'cpu4'
                         }
                     }
                     steps {
@@ -90,6 +88,7 @@ pipeline {
                     }
                     environment {
                         HOME = "${WORKSPACE}"
+                        PARALLEL_LEVEL = '10'
                     }
                     steps {
                         cleanWs(
@@ -114,9 +113,8 @@ pipeline {
                     }
                     agent {
                         docker {
-                            image 'gpuci/rapidsai:22.04-cuda11.5-devel-centos7-py3.8'
-                            label 'driver-495'
-                            args '--runtime "nvidia" -e "NVIDIA_VISIBLE_DEVICES=$EXECUTOR_NUMBER"'
+                            image 'gpuci/rapidsai-driver:22.04-cuda11.5-devel-centos7-py3.8'
+                            label 'cpu'
                         }
                     }
                     environment {

@@ -23,28 +23,20 @@ rm -rf ${MORPHEUS_ROOT}/.cache/ ${MORPHEUS_ROOT}/build/
 conda config --add pkgs_dirs /opt/conda/pkgs
 conda config --env --add channels conda-forge
 conda config --env --set channel_alias ${CONDA_CHANNEL_ALIAS:-"https://conda.anaconda.org"}
-mamba install -q -y -n base -c conda-forge "boa >=0.10"
-mamba create -q -y -n morpheus python=${PYTHON_VER}
+mamba env create -q -n morpheus -f ${MORPHEUS_ROOT}/docker/conda/environments/cuda${CUDA_VER}_dev.yml
 conda activate morpheus
 
 gpuci_logger "Installing CI dependencies"
 mamba env update -q -n morpheus -f ${MORPHEUS_ROOT}/docker/conda/environments/cuda${CUDA_VER}_ci.yml
+conda deactivate && conda activate morpheus
+
+gpuci_logger "Final Conda Environment"
+show_conda_info
 
 gpuci_logger "Check versions"
 python3 --version
 gcc --version
 g++ --version
-
-show_conda_info
-
-gpuci_logger "Installing dependencies"
-mamba env update -q -n morpheus -f ${MORPHEUS_ROOT}/docker/conda/environments/cuda${CUDA_VER}_dev.yml
-conda deactivate && conda activate morpheus
-
-gpuci_logger "Final Conda Environment"
-conda list
-
-gpuci_logger "Check cmake & ninja"
 cmake --version
 ninja --version
 

@@ -17,9 +17,16 @@
 gpuci_logger "Env Setup"
 source /opt/conda/etc/profile.d/conda.sh
 export MORPHEUS_ROOT=${MORPHEUS_ROOT:-$(git rev-parse --show-toplevel)}
-gpuci_logger "Procs: $(nproc)"
-gpuci_logger "Memory"
 
+# For non-gpu hosts nproc will correctly report the number of cores we are able to use
+# On a GPU host however nproc will report the total number of cores and PARALLEL_LEVEL
+# will be defined specifying the subset we are allowed to use.
+NUM_CORES=$(nproc)
+export PARALLEL_LEVEL=${PARALLEL_LEVEL:-${NUM_CORES}}
+gpuci_logger "Procs: ${NUM_CORES}"
+/usr/bin/lscpu
+
+gpuci_logger "Memory"
 /usr/bin/free -g
 
 gpuci_logger "User Info"

@@ -17,7 +17,6 @@
 set -e
 
 source ${WORKSPACE}/ci/scripts/jenkins/common.sh
-/usr/bin/nvidia-smi
 
 restore_conda_env
 pip install ${MORPHEUS_ROOT}/build/wheel
@@ -27,7 +26,7 @@ gpuci_logger "Installing Documentation dependencies"
 pip install -r requirement.txt
 
 gpuci_logger "Building docs"
-make html
+make -j ${PARALLEL_LEVEL} html
 
 gpuci_logger "Tarring the docs"
 tar cfj "${WORKSPACE_TMP}/docs.tar.bz" build/html

@@ -138,10 +138,10 @@ COPY docker/conda/environments/cuda${CUDA_VER}_runtime.yml ./docker/conda/enviro
 RUN --mount=type=bind,from=conda_bld_morpheus,source=/opt/conda/conda-bld,target=/opt/conda/conda-bld \
     --mount=type=cache,id=conda_pkgs,target=/opt/conda/pkgs,sharing=locked \
     source activate morpheus &&\
-    # Install runtime dependencies that are pip-only
-    /opt/conda/bin/mamba env update -n morpheus --file docker/conda/environments/cuda${CUDA_VER}_runtime.yml &&\
     # Install morpheus
     CONDA_ALWAYS_YES=true /opt/conda/bin/mamba install -n morpheus -c local -c rapidsai -c nvidia -c nvidia/label/dev -c conda-forge morpheus &&\
+    # Install runtime dependencies that are pip-only
+    /opt/conda/bin/mamba env update -n morpheus --file docker/conda/environments/cuda${CUDA_VER}_runtime.yml &&\
     # Clean and activate
     conda clean -afy
 

@@ -58,7 +58,7 @@ def accepted_types(self) -> typing.Tuple:
 
     def _postprocess(self, x: MultiPostprocLogParsingMessage):
 
-        infer_pdf = pd.DataFrame(x.seq_ids).astype(int)
+        infer_pdf = pd.DataFrame(x.seq_ids.get()).astype(int)
         infer_pdf.columns = ["doc", "start", "stop"]
         infer_pdf["confidences"] = x.confidences.tolist()
         infer_pdf["labels"] = x.labels.tolist()

@@ -3,7 +3,7 @@ cuml==0+untagged.1.g835a9ae
 dateparser==1.1.0
 matplotlib==3.4.3
 networkx==2.6.3
-numpy==1.19.5
+numpy==1.22.0
 pandas==1.3.3
 scikit_learn==1.0.2
 stellargraph==1.2.1

@@ -3,7 +3,7 @@ cuml==0+untagged.1.g835a9ae
 dateparser==1.1.0
 matplotlib==3.4.3
 networkx==2.6.3
-numpy==1.19.5
+numpy==1.22.0
 pandas==1.3.3
 scikit_learn==1.0.2
 stellargraph==1.2.1

@@ -3,7 +3,7 @@ cupy==9.1.0
 dfencoder==0.0.35
 dill==0.3.4
 matplotlib==3.4.2
-numpy==1.20.3
+numpy==1.22.0
 pandas==1.0.1
 scikit_learn==1.0.2
 torch==1.7.1
@@ -1,7 +1,7 @@
 clx==0.0.1
 cudf==0+untagged.1.g788bddd
 cupy==9.1.0
-numpy==1.20.3
+numpy==1.22.0
 onnxruntime==1.10.0
 s3fs==2021.5.0
 scipy==1.6.0