Consistent usage of 'cuda' or 'gpu'

[benjamin-kroeger]

Could you change the device specifications to one consistent name, either cuda cuda or gpu, although I would suggest using cuda since this works with pytorch as well.