gpu版tensorflow测试

测试程序：

import tensorflow as tfwith tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=False)) as sess:a = tf.constant(1)b = tf.constant(3)c = a + bprint('结果是：%d\n 值为：%d' % (sess.run(c), sess.run(c)))

输出显卡型号、内存等：

tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=False))

allow_soft_placement=True 表示使用不能使用显卡时使用cpu

log_device_placement=False 不打印日志，不然会刷屏

gpu版tensorflow测试相关推荐

Ubuntu 16.04 LTS下编译GPU版tensorflow
Ubuntu 16.04 LTS下编译GPU版tensorflow 机器学习与数学 · 2016-06-10 13:51 作者: 比特小组机器学习与数学出品机器学习必然涉及到代码,本小组选择sci ...
GTX1060安装gpu版tensorflow经验分享
配适版本作为一个深度学习小白,刚刚开始接触深度学习,没想到安装tensorflow竟然成为了第一支拦路虎-用CPU的版本完成作业后想着既然显卡还凑活不如用GPU版本的试试,没想到安装实在麻烦,网上的 ...
(转）在Windows上安装GPU版Tensorflow
转载自在Windows上安装GPU版Tensorflow. 1. 下载安装Anaconda 简单说就是下载 64位 python 3.5 版本的Anaconda https://www.continu ...
Win10安装GPU版tensorflow和keras
Win10安装GPU版tensorflow和keras 1.python安装在此选用python3.7安装见详细教程https://blog.csdn.net/weixin_43545253/art ...
错误调试：GPU 版 TensorFlow failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
如果你是使用 GPU 版 TensorFlow 的话,并且你想在显卡高占用率的情况下(比如玩游戏)训练模型,那你要注意在初始化 Session 的时候为其分配固定数量的显存,否则可能会在开始训练的时候 ...
conda安装gpu版tensorflow和局部cuda插件，不用配置全局cuda了
conda安装gpu版tensorflow和局部cuda插件,不用配置全局cuda了把后面的命令改为你要安装的版本号 conda install -c conda-forge tensorflow- ...
ubuntu16.04下安装GTX1080TI显卡驱动+安装CUDA+Cudnn+anaconda+GPU版tensorflow
参考博客:https://blog.csdn.net/weixin_40294256/article/details/79157838 先安装好GTX1080Ti 显卡驱动--安装CUDA--安装Cu ...
Win10通过Anaconda安装GPU版tensorflow
目录通过conda安装CUDA.cudnn方法不通过conda安装CUDA.cudnn方法安装前言: 安装VS2015 CUDA和Cudnn 安装Anaconda3 测试: 附录: 通过cond ...
GPU版TensorFlow怎么指定让CPU运行
由于某些原因GPU版的TensorFlow运行起来会出现一些问题,比如内存溢出等情况.此时我们可以用CPU和系统内存来运行我们的程序. 代码如下: import os os.environ[" ...

gpu版tensorflow测试

gpu版tensorflow测试相关推荐

最新文章

热门文章