rldurham.check_device

rldurham.check_device()

Check if CUDA is available and return ‘cuda’ or ‘cpu’ accordingly. Also prints warning/confirmation as it is recommended to use non-GPU environments.

Returns:

‘cuda’ or ‘cpu’