Might not detect GPUs out of the box

[kaimast]

The line below seems to limit which GPUs are used by the script. Maybe document that somewhere, or change it to include 0 and 1?

https://github.com/LaceyChen17/DREAM/blob/b49fee5c0be7a3fe06b5070aa61e1d8cd53f78b7/src/constants.py#L1