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