tensorflow获取可用运算设备(CPU,GPU)

最新推荐文章于 2024-10-02 11:45:00 发布
最新推荐文章于 2024-10-02 11:45:00 发布
阅读量8.8k 收藏 9
点赞数 1
分类专栏： 深度学习 随笔 文章标签： GPU
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/littlehaes/article/details/82317220
版权
本文介绍了在遇到TensorFlow只能使用CPU时的排查过程。问题在于系统同时安装了tensorflow和tensorflow-gpu两个包，导致GPU设备未被正确识别。通过卸载无GPU版本的tensorflow，然后重新安装tensorflow-gpu，最终成功启用GPU设备。
摘要由CSDN通过智能技术生成

Welcome To My Blog
正常来说,运行下面两句会打印tensorflow能用的CPU和GPU

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

结果发现只有CPU可用,查阅资料后发现可能跟tensorflow版本有关,在终端运行pip3 list查看安装的包

pip3 list

发现有两个tensorflow,一个是tensorflow,还有一个是tensorflow-gpu.卸载无gpu版本的tensorflow

pip3 uninstall tensorflow

重新运行最开始两句代码,结果报错:

ImportError: No module named '
最低0.47元/天 解锁文章
littlehaes
关注
专栏目录
tensorflow学习之路(5):利用tensorflow输出devices信息
IB_H20的博客
06-20 4189
tensorflow tf.device()
tensorflow查看电脑的CPU和GPU
修炼之路
06-09 4万+
1、查看电脑GPU和CPUimport os from tensorflow.python.client import device_lib os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99" if __name__ == "__main__": print(device_lib.list_local_devices())2、指定CPU或GPU进行计算 ...
使用tensorflow查看CPU信息
fsdgfsf的博客
05-23 445
我的tensorflow版本是1.8.0 用下面的博客中的代码是ok的 博客
深度学习笔记18_TensorFlow实现猫狗识别
最新发布
L1073203482的博客
10-02 1123
本周通过tenserflow框架创建VGG16网络模型进行猫狗识别：VGG16模型结构：VGG16共包含：13个卷积层（Convolutional Layer），分别用conv3-XXX表示 (XXX为输出通道数，3代表kernel_size)3个全连接层（Fully connected Layer）,分别用FC-XXXX表示（XXX为输出神经元个数)5个池化层（Pool layer）,分别用maxpool表示。VGG优缺点分析：VGG优点。
Tensorflow查看版本以及是否CPU版本
ssjdoudou的博客
09-10 9223
1、查看版本 Arithmetic@qingjiaowosuanshujiadeMacBook-Pro leetcode % python3 Python 3.6.6 (v3.6.6:4cf1f54eb7, Jun 26 2018, 19:50:54) [GCC 4.2.1 Compatible Apple LLVM 6.0 (clang-600.0.57)] on darwin Type ...
TensorFlow的CPU与GPU指定问题
GUEST_MTE的博客
02-02 1160
当同时安装CPU版本与GPU版本的TensorFlow时，默认是如果版本相同，默认是使用GPU版本的。
tf.device()指定tensorflow运行的GPU或CPU设备
热门推荐
牧野的博客
03-29 7万+
在tensorflow中，我们可以使用 tf.device() 指定模型运行的具体设备，可以指定运行在GPU还是CUP上，以及哪块GPU上。设置使用GPU使用 tf.device('/gpu:1') 指定Session在第二块GPU上运行：import tensorflow as tf with tf.device('/gpu:1'): v1 = tf.constant([1.0, 2....
tensorflow指定CPU与GPU运算的方法实现
09-17
在使用TensorFlow进行深度学习...通过这些方法，你可以灵活地控制TensorFlow在CPU和GPU之间的运算分配，以及GPU资源的使用，从而优化性能和资源利用率。在进行大规模模型训练或在多GPU环境中工作时，这些技巧尤其重要。
使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)
09-16
当设置为`True`时，`allow_soft_placement`允许TensorFlow自动将运算放置在可用的设备上，即使该运算原本是为GPU设计但当前GPU不可用。例如： ```python config = tf.ConfigProto() config.allow_soft_placement = ...
运行tensorflow python程序,限制对GPU和CPU的占用操作
12-20
默认情况下，TensorFlow会尝试利用系统中所有可用的GPU，这可能导致其他进程无法访问GPU资源，或者导致资源浪费。以下是如何限制TensorFlow对GPU和CPU的占用。 首先，针对GPU的占用控制，你可以通过设置环境变量`...
Tensorflow模型预测时，若用CPU计算，如何设置？ 并行计算-多核(multicore)，多线程(multi-thread)
郭云飞的专栏
07-05 1万+
在标准系统上通常有多个计算设备. TensorFlow 支持 CPU 和 GPU 这两种设备. 用指定字符串来标识这些设备. 比如: "/cpu:0": 机器中的 CPU "/gpu:0": 机器中的 GPU, 如果你有一个的话. "/gpu:1": 机器中的第二个 GPU, 以此类推... 在Tensorflow程序中，我们会经常看到 with tf.device("/cpu:0"): 这个语...
TensorFlow之设备(device)详解
荒野雄兵的专栏
04-18 8377
对TensorFlow不了解的，猛的一看还以为TensorFlow是个设备什么的，其实本文要说的是Python的TensorFlow模块下一个叫device()的方法，即: import tensorflow as tf with tf.device('/gpu:0'): ... ... ... 设备是指一块可以用来运算并且拥有自己的地址空间的硬件，如CPU和GPU T...
(转)tensorflow 使用CPU而不使用GPU的问题解决
zyb418的专栏
03-11 5628
https://www.cnblogs.com/hutao722/p/9583214.html 今天发现一个怪现象，在训练keras时，发现不使用GPU进行计算，而是采用CPU进行计算，导致计算速度很慢。 用如下代码可检测tensorflow的能使用设备情况： from tensorflow.python.client import device_lib print(device_li...
【DL-安装遇错】tensorflow突然检测不到GPU，显示False的问题
斯基的朦胧
01-11 3794
tensorflow突然检测不到GPU，显示False的问题 问题描述： 之前装好后输入一下命令，可以正常显示一下内容；但是之后重新安装了TensorFlow-GPU后测试，检测不到GPU，显示false的问题。 # 检测 tensorflow 能使用的设备情况 from tensorflow.python.client import device_lib #os.environ["CUDA_VISIBLE_DEVICES"] = "1" # 这个可以指定使用哪个设备 print(device_lib.
windows下使用tensorflow指定gpu运行程序
JustNo
05-22 3729
查看CPU/GPU信息: 输入以下命令: import os from tensorflow.python.client import device_lib os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99" if __name__ == "__main__": print(device_lib.list_local_devices()) 我们可...
关于anadonada中 tensorflow无法调用cpu的情况记录
a694262054的博客
03-25 861
这个垃圾的bug浪费了我一个下午的时间，最终在github中找到了对我这个问题的解决方法网址：https://github.com/tensorflow/tensorflow/issues/7648#issuecomment-280866214该bug主要表现为tensorflow无法使用gpu进行计算，而cuda和显卡驱动都是正常的请先按照上面的网找对照我的问题是anaconda中的tensor...
查看tensorflow cpu或GPU是否安装成功
Nancy_Nancy_li的博客
12-26 2776
tensorflow cpu或GPU
使用GPU加速TensorFlow进行MNIST数据训练
在TensorFlow中，可以通过tf.config.experimental.list_physical_devices('GPU')来检测系统中是否有可用的GPU。如果有多个GPU，还可以通过tf.config.experimental.set_memory_growth设置来指定使用哪一个GPU或者分配...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值