I am trying to add a custom c++ layer defining a new loss layer (central loss) for Caffe. The reason going with the c++ approach and not a python layer is for efficiency reasons (since python layers currently are limited to a single GPU). More specifically make use of https://raw.githubusercontent.com/ydwen/caffe-face
However, compared to the docker images with DIGITS or Nividia-Caffe for the DGX-1, they seem to use another version of Caffe. They make use of different functions and different templates etc. for the layers. If we look at Nividia-Caffe 0.15 (on GitHub) it complies with our center loss, while the experimental branch does not.
Is there any docker image using Nividia-Caffe 0.15 (or any other version which complies with the Caffe templates etc) available for the DGX-1?
Also, what is special with the docker images for the DGX-1 compared with the usual docker images publicly available on GitHub?