Reduce CUDA footprint in containers (aad86597) · Commits · 郑智淋 / lammps

tools/singularity/ubuntu18.04_amd_rocm_cuda.def

+16 −6

Original line number	Diff line number	Diff line
		@@ -2,11 +2,11 @@ BootStrap: library
		From: lammps/default/lammps_development:ubuntu18.04_amd_rocm

		%environment
		export PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
		export CUDADIR=/usr/local/cuda
		export CUDA_PATH=/usr/local/cuda
		export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
		export LIBRARY_PATH=/usr/local/cuda/lib64/stubs
		export PATH=/usr/local/cuda-10.2/bin:${PATH}:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin/x86_64
		export CUDADIR=/usr/local/cuda-10.2
		export CUDA_PATH=/usr/local/cuda-10.2
		export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64
		export LIBRARY_PATH=/usr/local/cuda-10.2/lib64/stubs

		%post
		export DEBIAN_FRONTEND=noninteractive
		@@ -15,7 +15,17 @@ From: lammps/default/lammps_development:ubuntu18.04_amd_rocm
		apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
		add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
		apt-get update
		apt-get -y install --no-install-recommends cuda cuda-compiler-10-2 cuda-compat-10-2

		export CUDA_PKG_VERSION=10.2

		apt-get install -y --no-install-recommends \
		cuda-libraries-$CUDA_PKG_VERSION \
		cuda-command-line-tools-$CUDA_PKG_VERSION \
		cuda-libraries-dev-$CUDA_PKG_VERSION \
		cuda-minimal-build-$CUDA_PKG_VERSION \
		cuda-compat-$CUDA_PKG_VERSION \
		libcublas10 \
		libcublas-dev


		echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf

tools/singularity/ubuntu18.04_gpu.def

+16 −9

Original line number	Diff line number	Diff line
		@@ -2,11 +2,11 @@ BootStrap: docker
		From: rocm/dev-ubuntu-18.04

		%environment
		export PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin/x86_64
		export CUDADIR=/usr/local/cuda
		export CUDA_PATH=/usr/local/cuda
		export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
		export LIBRARY_PATH=/usr/local/cuda/lib64/stubs
		export PATH=/usr/local/cuda-10.2/bin:${PATH}:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin/x86_64
		export CUDADIR=/usr/local/cuda-10.2
		export CUDA_PATH=/usr/local/cuda-10.2
		export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64
		export LIBRARY_PATH=/usr/local/cuda-10.2/lib64/stubs
		%post
		export DEBIAN_FRONTEND=noninteractive
		apt-get update
		@@ -73,10 +73,17 @@ From: rocm/dev-ubuntu-18.04
		apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
		add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
		apt-get update
		apt-get install --no-install-recommends -y \
		cuda \
		cuda-compiler-10-2 \
		cuda-compat-10-2

		export CUDA_PKG_VERSION=10.2

		apt-get install -y --no-install-recommends \
		cuda-libraries-$CUDA_PKG_VERSION \
		cuda-command-line-tools-$CUDA_PKG_VERSION \
		cuda-libraries-dev-$CUDA_PKG_VERSION \
		cuda-minimal-build-$CUDA_PKG_VERSION \
		cuda-compat-$CUDA_PKG_VERSION \
		libcublas10 \
		libcublas-dev

		export PATH=$PATH:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin/x86_64
		git clone -b master-rocm-3.3 https://github.com/ROCmSoftwarePlatform/hipCUB.git

Admin message