Как определить, использует ли тензорный поток ускорение графического процессора изнутри оболочки Python? ⇐ Python

[Anonymous]

Я установил tensorflow в свою Ubuntu 16.04, используя второй ответ здесь со встроенной установкой apt cuda в Ubuntu.

Теперь мой вопрос: как я могу проверить, действительно ли tensorflow использует графический процессор? У меня графический процессор GTX 960M. Когда я импортирую тензорный поток, это результат

Код: Выделить всё

I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcurand.so locally

Достаточно ли этих выводов, чтобы проверить, использует ли тензорный поток графический процессор?

Подробнее здесь: https://stackoverflow.com/questions/380 ... thon-shell