An open API service indexing awesome lists of open source software.

https://github.com/dmalexx/cuda_check

How can you check if CUDA is available in Tensorflow
https://github.com/dmalexx/cuda_check

cuda python tensorflow

Last synced: 3 months ago
JSON representation

How can you check if CUDA is available in Tensorflow

Awesome Lists containing this project

README

          

- If TensorFlow does not find a GPU, the output will be:

CUDA Available: True # or False if TensorFlow is not built with CUDA support

GPU Available: []

- Output (when everything is working correctly):

CUDA Available: True

GPU Available: [PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]