NVidia/AMD OpenCL Docker base image

This is a basic Docker image, capable of running OpenCL containers both on NVidia and AMD GPUs The image is based on nvidia/opencl image with additional AMD OpenCL libraries

Usage

Running with NVidia GPUs

Requirements:

  • NVidia Docker 2 installed
docker run --runtime=nvidia --rm -it sonm/opencl clinfo

Running with AMD GPUs

Requirements:

  • AMD drivers installed
docker run --device /dev/dri:/dev/dri --rm -it sonm/opencl clinfo