small fix (multiple gpus only via uncomment)