Some tutorials are geared towards GPUs, but the tutorials should be running on as much hardware as possible. I think we can easily solve this by using the global config heat.devices.use_device('GPU') instead of passing device='gpu' to all array calls. Then we just put this by itself in a Jupyter notebook cell and if this fails because we don't have a gpu, we just carry on on CPU.
Some tutorials are geared towards GPUs, but the tutorials should be running on as much hardware as possible. I think we can easily solve this by using the global config
heat.devices.use_device('GPU')instead of passingdevice='gpu'to all array calls. Then we just put this by itself in a Jupyter notebook cell and if this fails because we don't have a gpu, we just carry on on CPU.