/mask-gpu

A simple tool to expose only specified number of GPUs with desired memory to Tensorflow

Primary LanguagePythonMIT LicenseMIT

Watchers