casperhansen's picture
Set initial device to cuda:0 for faster initial loading
364bcfc