gpt4 book ai didi

python - 无法在多个 GPU 上运行 tensorflow

转载 作者:行者123 更新时间:2023-11-28 16:27:56 26 4
gpt4 key购买 nike

我正在运行来自 tensorflow 存储库的 cifar10 多 GPU 示例。我能够使用多个 GPU。我的 ubuntu PC 有两个 Titan X,我看到两个 GPU 上的进程完全占用了内存。然而,实际上只有一个 GPU 在计算。我没有得到任何加速。我已经尝试过 tensorflow 0.5.0 和 0.6.0 pip 二进制文件。我也试过从源代码编译。

编辑:安装旧版本的 nvidia 驱动程序后,问题消失了。

最佳答案

在我安装了旧版本(352.55)的 nvidia 驱动程序后,问题消失了。

关于python - 无法在多个 GPU 上运行 tensorflow,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34860281/

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