TensorFlow supports running computations on a variety of types of devices, including CPU and GPU.
Is TensorFlow CPU or GPU?
TensorFlow 2 has finally became available this fall and as expected, it offers support for both standard CPU as well as GPU based deep learning.
How can I tell if TensorFlow is using my GPU?
Recently a few helpful functions appeared in TF:
- tf. test. is_gpu_available tells if the gpu is available.
- tf. test. gpu_device_name returns the name of the gpu device.
