Commit e7a8de03 authored by Richard Berger's avatar Richard Berger
Browse files

Add container with both ROCm and CUDA

parent 1c94415d
Loading
Loading
Loading
Loading
+25 −0
Original line number Diff line number Diff line
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