This works for cog build but not cog run and cog predict. There's a little thing that checks for the error returned, and re-runs it without --gpus.

Requested by @charlesfrye.