Features

Fixes

  • Pass gpu count to avoid bad nvidia-smi call