Allow specification of number of GPU devices