39 lines
1.4 KiB
Modula-2
39 lines
1.4 KiB
Modula-2
BootStrap: library
|
|
From: lammps/default/lammps_development:ubuntu18.04_amd_rocm
|
|
|
|
%environment
|
|
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
|
|
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
|
|
mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
|
|
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
|
|
|
|
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
|
|
echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
|
|
|
|
# clean cache
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
%labels
|
|
Author rbberger
|