docker: switch nvidia build to bash for hpcx

devitocodes · May 1, 2024 · a400992 · a400992
1 parent fe4ec46
commit a400992
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docker/Dockerfile.nvidia b/docker/Dockerfile.nvidia
@@ -9,6 +9,8 @@ ARG arch="nvc"
 ########################################################################
 FROM ubuntu:22.04 as sdk-base
 
+SHELL ["/bin/bash", "-c"]
+
 ENV DEBIAN_FRONTEND noninteractive
 
 # Install python
@@ -137,7 +139,7 @@ FROM sdk-base as nvc
 ADD docker/nvdashboard.json /app/nvdashboard.json
 
 # mpi4py
-ENV MPI4PY_FLAGS='. $HPCSDK_HOME/comm_libs/hpcx/latest/hpcx-init.sh && hpcx_load && CC=nvc CFLAGS="-noswitcherror -tp=px"'
+ENV MPI4PY_FLAGS='source $HPCSDK_HOME/comm_libs/hpcx/latest/hpcx-init.sh && hpcx_load && CC=nvc CFLAGS="-noswitcherror -tp=px"'
 
 ENV DEVITO_ARCH="nvc"
 ENV DEVITO_PLATFORM="nvidiaX"