gpt4 book ai didi

pytorch - 如何告诉 PyTorch 不使用 GPU?

转载 作者:行者123 更新时间:2023-12-04 17:06:44 25 4
gpt4 key购买 nike

我想做一些 CPU 和 GPU 之间的时序比较以及一些分析,想知道是否有办法告诉 不使用 GPU 而是只使用 CPU?我意识到我可以安装另一个仅限 CPU 的 ,但希望有更简单的方法。

最佳答案

您可以设置 CUDA_VISIBLE_DEVICES在运行您的 Torch 代码之前通过 shell 清空变量。export CUDA_VISIBLE_DEVICES=""应该告诉 torch 没有 GPU。export CUDA_VISIBLE_DEVICES="0"会告诉它只使用一个 GPU(id 为 0 的那个)等等。

关于pytorch - 如何告诉 PyTorch 不使用 GPU?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53266350/

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