Hi @alie89
Please follow this discussion GPU Access Limited by CGroup - #6 by Phhere
In a nutshell - you would need cgroups , also with the help of it you can map the topology to a bit.
I am not sure whether there are controlling features available related to gpu memory , gpu cores etc.