Update Ubuntu 18.04 GPU container (af2e1bf7) · Commits · 郑智淋 / lammps

tools/singularity/ubuntu18.04_gpu.def

+19 −4

Original line number	Diff line number	Diff line
		BootStrap: docker
		From: rocm/dev-ubuntu-18.04
		From: ubuntu:18.04

		%environment
		export PATH=/usr/lib/ccache:/usr/local/cuda-10.2/bin:${PATH}:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin/x86_64
		@@ -11,6 +11,22 @@ From: rocm/dev-ubuntu-18.04
		export DEBIAN_FRONTEND=noninteractive
		apt-get update
		apt-get upgrade --no-install-recommends -y

		apt-get install -y --no-install-recommends curl libnuma-dev gnupg

		curl -sL http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key \| apt-key add -
		printf "deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main" > /etc/apt/sources.list.d/rocm.list

		apt-get update
		apt-get install --no-install-recommends -y \
		kmod \
		file \
		sudo \
		libelf1 \
		rocm-dev \
		rocm-libs \
		build-essential

		apt-get install --no-install-recommends -y software-properties-common
		apt-get install --no-install-recommends -y \
		bc \
		@@ -57,7 +73,6 @@ From: rocm/dev-ubuntu-18.04
		python3-pkg-resources \
		python3-setuptools \
		python3-virtualenv \
		rocm-libs \
		rsync \
		ssh \
		vim-nox \
		@@ -86,10 +101,10 @@ From: rocm/dev-ubuntu-18.04
		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
		git clone -b master-rocm-3.5 https://github.com/ROCmSoftwarePlatform/hipCUB.git
		mkdir hipCUB/build
		cd hipCUB/build
		CXX=hcc cmake -D BUILD_TEST=off ..
		CXX=hipcc cmake -D BUILD_TEST=off ..
		make -j4
		make package
		make install