首页 > 系统服务 > 详细

tensorflow-gpu版linux端安装成功检验

时间:2020-05-09 22:30:56      阅读:134      评论:0      收藏:0      [点我收藏+]

检验代码:

import tensorflow as tf
a=tf.test.is_built_with_cuda()
b=tf.test.is_gpu_available(cuda_only=False,min_cuda_compute_capability=None)
print(a)
print(b)

执行这段代码应该显示为:

技术分享图片

技术分享图片

 

检验代码2:

import tensorflow as tf
hello = tf.constant(Hello, TensorFlow!)
sess = tf.Session()
print(sess.run(hello))

安装成功的正确结果应该为:

技术分享图片

技术分享图片

 

验证环境:

技术分享图片

技术分享图片

tensorflow-gpu1.15.0

tensorflow-gpu版linux端安装成功检验

原文:https://www.cnblogs.com/acm-icpcer/p/12860163.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!