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

最新推荐文章于 2024-08-13 13:48:39 发布
最新推荐文章于 2024-08-13 13:48:39 发布
阅读量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查看电脑的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 433
我的tensorflow版本是1.8.0 用下面的博客中的代码是ok的 博客
Tensorflow-gpu安装教程（Win11, Anaconda3，Python3.9）
最新发布
2401_86489187的博客
08-13 1192
CUDA并不是实现GPU的神经网络加速库，如果希望针对的是神经网络进行加速，我们还需要安装cuDNN神经网络加速库。第一次单击下载时，会让你先注册登录，然后再进行下载，注册过程认真填写内容就没问题，此处略过，接下来进入下载环节。完成安装后，检查一下环境变量是否存在，一般安装完成会自动配置好环境变量，若是没有，则需手动配置，具体过程如下。查看是否有以下系统变量，没有则需要自行添加，对应图片上的名称和值，配置你电脑CUDA安装所在的位置。，不然很容易安装失败，或者就按官网查询的组合安装，安装过程是一样的！
Tensorflow查看版本以及是否CPU版本
ssjdoudou的博客
09-10 9207
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 997
当同时安装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 8266
对TensorFlow不了解的，猛的一看还以为TensorFlow是个设备什么的，其实本文要说的是Python的TensorFlow模块下一个叫device()的方法，即: import tensorflow as tf with tf.device('/gpu:0'): ... ... ... 设备是指一块可以用来运算并且拥有自己的地址空间的硬件，如CPU和GPU T...
tensorflow学习之路(5):利用tensorflow输出devices信息
IB_H20的博客
06-20 4165
tensorflow tf.device()
【DL-安装遇错】tensorflow突然检测不到GPU，显示False的问题
斯基的朦胧
01-11 3754
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 3702
查看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 846
这个垃圾的bug浪费了我一个下午的时间，最终在github中找到了对我这个问题的解决方法网址：https://github.com/tensorflow/tensorflow/issues/7648#issuecomment-280866214该bug主要表现为tensorflow无法使用gpu进行计算，而cuda和显卡驱动都是正常的请先按照上面的网找对照我的问题是anaconda中的tensor...
tensorflow-gpu与tensorflow有什么区别
06-25
4. **安装依赖**：可能需要单独安装 CUDA 和 cuDNN（GPU 加速运算库）才能使用 TensorFlow-GPU。 如果你正在考虑使用 TensorFlow，你可能需要根据你的应用场景和可用资源来决定是使用标准版还是 GPU 版本。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值