d

Signed-off-by: [email protected] <[email protected]>
spidernet-io · Jul 30, 2024 · d7c9d5b · d7c9d5b
1 parent c1a5e69
commit d7c9d5b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/rdma-tools/image/Dockerfile.template b/rdma-tools/image/Dockerfile.template
@@ -71,6 +71,9 @@ RUN chmod +x /install-tools.sh && /install-tools.sh && rm -f /install-tools.sh
 # check binary
 RUN chmod +x /test.sh && /test.sh && rm -f /test.sh
 
+USER root
+RUN ulimit -l 2000000 && sudo service ssh start
+
 # Begin auto-generated paths
 ENV HPCX_DIR=/opt/hpcx
 ENV HPCX_UCX_DIR=/opt/hpcx/ucx
@@ -100,7 +103,4 @@ ENV CPATH=/opt/hpcx/ompi/include:/opt/hpcx/ucc/include:/opt/hpcx/ucx/include:/op
 ENV PKG_CONFIG_PATH=/opt/hpcx/hcoll/lib/pkgconfig:/opt/hpcx/sharp/lib/pkgconfig:/opt/hpcx/ucx/lib/pkgconfig:/opt/hpcx/ompi/lib/pkgconfig:
 # End of auto-generated paths
 
-
-RUN sudo ulimit -l 2000000 && sudo service ssh start
-
 CMD ["/entry.sh"]
diff --git a/rdma-tools/image/generateDockerfile.sh b/rdma-tools/image/generateDockerfile.sh
@@ -22,7 +22,6 @@ export ENV_DOWNLOAD_HPCX_URL="https://content.mellanox.com/hpc/hpc-x/${ENV_VERSI
 
 export ENV_DOWNLOAD_OFED_DEB_SOURCE="https://linux.mellanox.com/public/repo/mlnx_ofed/latest/ubuntu22.04/mellanox_mlnx_ofed.list"
 
-
 # NCCL 2.22.3, for CUDA 12.5, ubuntu 22.04
 # https://developer.nvidia.com/cuda-downloads
 export ENV_CUDA_DEB_SOURCE="https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb"