当前位置: 首页 > 知识库问答 >
问题:

选择GPU tenstorflow Directml或多图形处理器

袁桐
2023-03-14

我正在Windows PC上使用tensorflow训练一个模型,但是训练量很低,所以我尝试配置tensorflow以使用GPU。我安装了tensorflow directml(在使用python 3.6的conda环境中),因为我的GPU是AMD Radeon GPU。用这个简单的代码

import tensorflow as tf
tf.test.is_gpu_available()

我收到这个输出

2021-05-14 11:02:30.113880:I tensorflow/core/platform/cpu_feature_guard.cc:142]您的cpu支持未编译此tensorflow二进制文件以使用的指令:AVX2 2021-05-14 11:02:30.121580:I tensorflow/stream_executor/platform/default/dso_loader.cc:99]已成功打开动态库C:\Users\v.rocca\anaconda3\envs\tfradeon\lib\site packages\tensorflow\u core\python/directml.adbd007a01a52364381a1c71ebb6fa1b2389c88d.dll 2021-05-14 11:02:30.765470:I tensorflow/core/common\u runtime/dml/dml\u device\u cache.cc:249]directml设备枚举:找到2个兼容的适配器。2021-05-14 11:02:30.984834:I tensorflow/core/common_runtime/dml/dml_device_cache.cc:185]DirectML:在适配器0上创建设备(Radeon(TM)530)2021-05-14 11:02:31.150992:I tensorflow/stream_executor/platform/default/dso_loader.cc:99]成功打开动态库Kernel32.dll 2021-05-14 11:02:31.174716:I tensorflow/core/common_runtime/dml/dml_device_cache.cc:185]DirectML:在适配器1上创建设备(英特尔(R)UHD Graphics 620)True

因此,tensorflow采用集成的英特尔GPU,而不是Radeon GPU。如果我从管理硬件禁用Intel GPU,我将在输出中收到正确的GPU

2021-05-14 10:47:09.171568:我的TensorFlow/核心/平台/cpu_feature_guard.cc:142]您的CPU支持此TensorFlow二进制文件未编译使用的指令: AVX2 2021-05-14 10:47:09.176828:我的TensorFlow/stream_executor/平台/默认/dso_loader.cc:99]成功打开动态库C:\用户\v.rocca\anaconda3\envs\tfradeon\lib\site-包\tensorflow_core\python/directml.adbd007a01a52364381a1c71ebb6fa1b2389c88d.dll2021-05-14 10:47:09.421265: I tenorflow/core/common_runtime/dml/dml_device_cache.cc:249]DirectML设备枚举:找到1个兼容的适配器。2021-05-14 10:47:09.626567: ITensorFlow/core/common_runtime/dml/dml_device_cache.cc:185]DirectML:在适配器0上创建设备(Radeon(TM)530)

我不想每次都禁用英特尔图形处理器,所以这是我的问题。是否可以选择我想要使用的GPU?或者可以同时使用两个GPU吗?谢啦

共有1个答案

万坚壁
2023-03-14

来自微软:

gpu_config = tf.GPUOptions()
gpu_config.visible_device_list = "1"

session = tf.Session(config=tf.ConfigProto(gpu_options=gpu_config))
 类似资料:
  • 目前我正在研究聊天服务器/客户端项目。我正在努力使用 select 处理多个请求,我的服务器脚本使用 select 模块,但客户端脚本没有。结果是,当用户输入消息时,其他客户端必须编写自己的消息才能通读对话。我已经在网上搜索了很多示例,但只能找到带有 sys.stdin 的代码片段,这不是我想要的。 我很高兴收到任何指示/解释。 服务器代码: 客户端代码:

  • 问题内容: 等待(不旋转)直到两个(多处理)队列中的任何一个都可用的最佳方法是什么(两者都驻留在同一系统上)? 问题答案: 似乎还没有一种正式的方式来解决这个问题。或至少不是基于此: http://bugs.python.org/issue3831 您可以尝试类似本文所进行的操作-访问基础管道文件句柄: http://haltcondition.net/?p=2319 然后使用选择。

  • 本文向大家介绍Java设计图形与多媒体处理,包括了Java设计图形与多媒体处理的使用技巧和注意事项,需要的朋友参考一下 本文实现了两个效果: 第一种,同心圆效果图: 第二种,播放音乐和切换图片的小程序效果图:  以上就是关于Java的图形设计以及多媒体处理的全部内容,希望对大家的学习有所帮助。

  • 编辑:我调试了应用程序,并用初始化了。这消除了错误,但是现在ImageView没有得到更新,但是当我从Gallery中选择image时,它已经更新了。

  • 5.2.4 图形的事件处理 面向对象的概念是和事件驱动编程联系在一起的。所谓事件是指在程序执行过程中发生的事情,例如点击了鼠标左键、按下了键盘上的回车键之类。某个对象可以与特定事件绑定 在一起,这样当特定事件发生时,可以调用特定的函数来处理这个事件。 画布及画布上的图形都是对象,都可以与交互事件绑定,这样用户可以利用键盘、鼠标 来操作、控制画布和图形。第 8 章将详细介绍 Tkinter 的事件驱

  • 问题内容: 我是Django的新手,如果您能向我提供一些指导,我将不胜感激。我正在尝试创建一种允许用户在一个或多个选项上打勾的表格。我知道我必须将MultiChoiceField字段与CheckboxSelectMultiple小部件一起使用,但是Django文档没有提供有关此主题的示例。如果您能给我一个例子并解释如何处理结果,将不胜感激。例如,如果我有一个带有选项abcd的表单,并且用户在c和d