Argh my mistake! I thiught we were able to get it in, but it turns out we fell back to just using the NVidia docker work. You can achieve this in other ways (using the experimental pluggable executors), but not in a container.