26 lines
1014 B
Modula-2
26 lines
1014 B
Modula-2
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
|
|
|
|
%post
|
|
|
|
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
|
|
apt-get -y install cuda cuda-compiler-10-2
|
|
|
|
|
|
echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf
|
|
echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
|
|
|
|
%labels
|
|
Author rbberger
|