I am running on work laptop not using GPU. (I'm running in docker). I just get
warnings.warn("FP16 is not supported on CPU; using FP32 instead")
And it works.