gpt4 book ai didi

tensorflow - 如何在Python接口(interface)中指定GPU的数量?

转载 作者:行者123 更新时间:2023-12-02 22:32:37 30 4
gpt4 key购买 nike

当我开始tensorflowSession 它立即在我的桌面上的两个 GPU(设备 0 和 2)上分配 100% 的内存。

如何使用 Python API将其限制为只有 1 个 GPU?

最佳答案

例如,将其限制为仅使用 GPU #0 的一种粗略方法是定义以下内容:

export CUDA_VISIBLE_DEVICES=0

关于tensorflow - 如何在Python接口(interface)中指定GPU的数量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33727116/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com