我想在 CPU 和 GPU 之间进行一些时序比较以及一些分析,并且想知道是否有办法区分pytorch /questions/tagged/pytorch不使用GPU而只使用CPU?我意识到我可以安装另一个仅 CPUpytorch /questions/tagged/pytorch,但希望有更简单的方法。
在运行代码之前,运行以下 shell 命令来告诉 torch 没有 GPU:
export CUDA_VISIBLE_DEVICES=""
这将告诉它仅使用一个 GPU(id 为 0 的 GPU),依此类推:
export CUDA_VISIBLE_DEVICES="0"
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)