ALPENGLOW Lab

Aerospace Lab of PENG for Leaders Of the World
Asst. Prof. Hao Peng, Embry-Riddle Aeronautical University, FL

User Tools

Site Tools


kb:programming:tensorflow_related

Tensorflow Related

Useful testing functions after setting up the environment:

  • Run import tensorflow as tf first
  • Run tf.test.gpu_device_name() should gives the correct GPU card name and driver info.
  • Run tf.test.is_built_with_cuda() tells if current TF is compiled supporting CUDA.
  • Check this link for more details: https://www.tensorflow.org/api_docs/python/tf/test

Some official Tested build configurations. It tells the compatibility among tensorflow, cuDNN, and CUDA (the package name cudatoolkit).

  • tensorflow=2.9.0, cuDNN=8.1, CUDA=11.2
kb/programming/tensorflow_related.txt · Last modified: 2024/07/31 16:10 by hao