Docker - opencl?

I need to run my target using opencl. Unfortunately, while docker (https://hub.docker.com/u/tvmai) shows the image tvmai/demo-opencl as existing, when I use bash.sh to try and launch it I get: (I can run demo-cpu just fine.)

  • docker: Error response from daemon: manifest for tvmai/demo-opencl:latest not found: manifest unknown: manifest unknown.

Okay, so now I’m thinking to try and re-compile for OpenCL. in config.cmake set(USE_OPENCL ON) - and ‘cmake …’, I get the error:

  • Could NOT find OpenCL (missing: OpenCL_LIBRARY OpenCL_INCLUDE_DIR)

Where is OpenCL on the demo-cpu image? Any ideas on how to install?

Thanks!!!