Created by: sneaxiy
This PR adds FLAGS_gpu_memory_limit_mb
(uint64) to limit the maximum gpu usage.
- If
FLAGS_gpu_memory_limit_mb > 0
, the gpu memory would be limit even though there is enough available memory. - If
FLAGS_gpu_memory_limit_mb = 0
, the all gpu memory is available to the process.