Limiting GPU memory usage in a container

Hi,

Suppose you can do this in a similar way as the below comment:

Thanks.