问如何选择带有TensorFlow和Yolov3的特定图形处理器？(tensorflow-gpu-2.3.1)
EN

Stack Overflow用户

提问于 2020-11-22 23:32:32

回答 1查看 2.4K关注 0票数 0

我的系统中有两个GPUS (RTX 2070)。我正在尝试根据下面的tutorial使用TensorFlow来应用Yolov3。

然而，TensorFlow似乎不能使用特定的图形处理器。我在互联网上搜索过，发现您需要在以下代码行中将'GPU'更改为/GPU:0：

physical_devices = tf.config.experimental.list_physical_devices('GPU')
if len(physical_devices) > 0:
tf.config.experimental.set_memory_growth(physical_devices[0], True)

然而，这不起作用，并不断给我以下错误：

raise ValueError("Memory growth cannot differ between GPU devices")
ValueError: Memory growth cannot differ between GPU devices

关于如何解决这个问题，有什么建议吗？

tensorflow

yolo

回答 1

Stack Overflow用户

发布于 2020-11-23 01:29:50

加上这个，每个索引范围(0，1)就是GPU：

tf.config.set_visible_devices(physical_devices[0:1], 'GPU')

票数 2

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/64956013

复制

相似问题

问如何选择带有TensorFlow和Yolov3的特定图形处理器？(tensorflow-gpu-2.3.1)
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问如何选择带有TensorFlow和Yolov3的特定图形处理器？(tensorflow-gpu-2.3.1)EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问如何选择带有TensorFlow和Yolov3的特定图形处理器？(tensorflow-gpu-2.3.1)
EN