Google Cloud Batch has accelerator optimized VMs, that need to be configured a little differently. They don't need to specify the accelerator type and count, but would still need the installGpuDrivers set to true. See Google documentation.
Suggest implementation
For this line, we also need to check if the task's machine type is g2-*, a2-* or a3-* even if accelerator type and count were not set. I think this should be enough. Happy to send a PR.
New feature
Usage scenario
Google Cloud Batch has accelerator optimized VMs, that need to be configured a little differently. They don't need to specify the accelerator type and count, but would still need the
installGpuDrivers
set to true. See Google documentation.Suggest implementation
For this line, we also need to check if the task's machine type is
g2-*
,a2-*
ora3-*
even if accelerator type and count were not set. I think this should be enough. Happy to send a PR.