gpt4 book ai didi

tensorflow - TensorFlow 在 CPU/GPU 之间切换

转载 作者:行者123 更新时间:2023-12-03 08:09:16 24 4
gpt4 key购买 nike

安装了 tensorflow GPU(在微不足道的 NVIDIA GeForce 950 上运行)后,我想将性能与 CPU 进行比较。

我正在运行 tensorFlow MNIST 教程代码,并注意到速度显着提高 - 无论如何估计(我两天前在批量大小为 100 的笔记本电脑 i7 上运行 CPU 版本,而这在桌面 GPU 上运行,批量大小10)--当我切换时在 CPU 和 GPU 之间......但我只注意到当我将 GPU 上的批量大小从 100 降低到 10 时速度增加......

现在我缺乏客观的衡量标准来衡量我的收获。

有没有办法在 CPU 和 GPU tensorflow 之间切换?

最佳答案

使 GPU 不可见

export CUDA_VISIBLE_DEVICES=""

恢复正常
unset CUDA_VISIBLE_DEVICES

关于tensorflow - TensorFlow 在 CPU/GPU 之间切换,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36584907/

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