From 14f691f0afbd14d93106672a6a7539b8376196f3 Mon Sep 17 00:00:00 2001 From: Richard Berger Date: Mon, 30 Nov 2020 14:46:24 -0500 Subject: [PATCH] Add support for NVIDIA OpenCL in Ubuntu gpu containers --- tools/singularity/ubuntu18.04_gpu.def | 8 ++++++++ tools/singularity/ubuntu20.04_gpu.def | 8 ++++++++ 2 files changed, 16 insertions(+) diff --git a/tools/singularity/ubuntu18.04_gpu.def b/tools/singularity/ubuntu18.04_gpu.def index c6a87a1ab9..cfd94c8531 100644 --- a/tools/singularity/ubuntu18.04_gpu.def +++ b/tools/singularity/ubuntu18.04_gpu.def @@ -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 diff --git a/tools/singularity/ubuntu20.04_gpu.def b/tools/singularity/ubuntu20.04_gpu.def index c0e713d8f1..29938b33f4 100644 --- a/tools/singularity/ubuntu20.04_gpu.def +++ b/tools/singularity/ubuntu20.04_gpu.def @@ -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