我想在CPU和GPU之间进行一些时间比较和性能分析，我想知道是否有办法告诉pytorch不使用GPU而只使用CPU？我知道我可以安装另一个仅CPU版本的pytorch，但希望有更简单的方法。

回答：

在运行您的代码之前，运行以下shell命令以告知torch没有可用的GPU：

export CUDA_VISIBLE_DEVICES=""

这将告诉它只使用一个GPU（id为0的那个），依此类推：

export CUDA_VISIBLE_DEVICES="0"