No it does not. In fact I had to make some modifications to the caffe2 source so it would even compile (had error no member named 'set_device_id' in 'caffe2::DeviceOption'). I’m guessing you haven’t actually tried to use any of your own models, I’ve found comments talking about how even when the demo worked using other models does not work. This twitter thread was quite revealing 8.