Visual Studio 2019下配置 CUDA 10.1 + TensorFlow-GPU 1.14.0

6. 验证测试

在工程 .py 文件中输入代码：

import tensorflow as tf
import os

#os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

greeting = tf.constant('Hello Google Tensorflow!')
sess = tf.compat.v1.Session()
result = sess.run(greeting)
print(result)
sess.close()

若控制台输出 GPU 相关信息 以及代码的输出信息 b'Hello Google Tensorflow!'，则环境搭建成功！

Note： tensorflow_gpu-1.14.0 中弃用了部分代码的接口，改用新的接口，如 tf.Session() 改为 tf.compat.v1.Session()，tf.placeholder 改为 tf.compat.v1.placeholder 。

到此这篇关于Visual Studio 2019下配置 CUDA 10.1 + TensorFlow-GPU 1.14.0的文章就介绍到这了,更多相关Visual Studio 2019配置CUDA TensorFlow-GPU内容请搜索易采站长站以前的文章或继续浏览下面的相关文章希望大家以后多多支持易采站长站！