测试tensorflow-gpu
测试 Tensorflow GPU
2023-09-14 09:05:38 时间
import tensorflow as tf
with tf.device('/gpu:0'):
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
with tf.Session() as sess:
print (sess.run(c))
相关文章
- LARK BOARD开发板试用第一篇-上电测试学习
- 权限系统(第一次测试)
- 使用ab和wrk对OSS进行benchmark测试
- 133、 Android 自动化测试(转载)
- linux(centos8):jmeter5.3并发测试实例(参数在范围内随机取值)
- Spring测试类代码编写
- TensorFlow:深度学习框架TensorFlow/TensorFlow_GPU的简介、安装、测试之详细攻略
- AI模型设计:Ubuntu18.04完美编译在阿里云镜像源tensorflow C++并实现深度学习计算【编译方法与测试深度学习C++源码已开源】
- 利用LabVIEW开发超声测试系统
- 面试官问:“给你一支笔,怎么测试?”这答案让人惊叹…
- Appium自动化测试基础--补充:C/S架构和B/S架构说明
- 【2023最全教程】Python接口测试实战之接口测试工具的使用(建议收藏)
- RTX30系列linux+docker容器的GPU配置(tensorflow-gpu==1.15~2.x、tensorrt 7、cuda、cudnn)附加resnet50模型测试
- [Linux] 使用 Mailx 发送测试(邮件基于25、465端口配置,云服务器)
- Airtest自动化测试篇——python编写和执行用例(下)