jianmu
/
lammps
forked from lijiext/lammps
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add support for NVIDIA OpenCL in Ubuntu gpu containers

This commit is contained in:
Richard Berger 2020-11-30 14:46:24 -05:00
parent 0f27ba34b0
commit 14f691f0af
No known key found for this signature in database
GPG Key ID: A9E83994E0BA0CAB
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tools/singularity/ubuntu18.04_gpu.def
View File

@ -69,6 +69,8 @@ From: ubuntu:18.04
mpi-default-bin \
mpi-default-dev \
ninja-build \
ocl-icd-libopencl1 \
ocl-icd-opencl-dev \
python-dev \
python-pip \
python-pygments \
@ -120,6 +122,12 @@ From: ubuntu:18.04
ln -s /usr/local/cuda-11.0 /usr/local/cuda
ln -s /usr/local/cuda-11.0/lib64/stubs/libcuda.so /usr/local/cuda-11.0/lib64/stubs/libcuda.so.1
###########################################################################
# NVIDIA OpenCL
###########################################################################
mkdir -p /etc/OpenCL/vendors
echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd
###########################################################################
# ROCm hipCUB

8
tools/singularity/ubuntu20.04_gpu.def
View File

@ -70,6 +70,8 @@ From: ubuntu:20.04
mpi-default-bin \
mpi-default-dev \
ninja-build \
ocl-icd-libopencl1 \
ocl-icd-opencl-dev \
python3-dev \
python3-pip \
python3-pkg-resources \
@ -117,6 +119,12 @@ From: ubuntu:20.04
ln -s /usr/local/cuda-11.0 /usr/local/cuda
ln -s /usr/local/cuda-11.0/lib64/stubs/libcuda.so /usr/local/cuda-11.0/lib64/stubs/libcuda.so.1
###########################################################################
# NVIDIA OpenCL
###########################################################################
mkdir -p /etc/OpenCL/vendors
echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd
###########################################################################
# ROCm hipCUB