Pytorch 硬件要求

【问题标题】：Pytorch Hardware RequirementPytorch 硬件要求
【发布时间】：2020-02-28 12:22:31
【问题描述】：

最新的 PyTorch 版本要求的最低计算能力是多少？

我有 Nvidia Geforce 820M，计算能力为 2.1。如何在我的 GPU 上运行 PyTorch 模型(if it doesn't support naturally)

【问题讨论】：

这是一个非常有趣的问题，虽然它有点离题（硬件）。 GCP 不是您要寻找的答案，但您知道什么时候需要。

【解决方案1】：

看看this page，PyTorch（即使是最古老的版本）从版本7.5 向上支持CUDA。鉴于this page，CUDA 7.5 需要最低计算能力2.0。因此，在纸面上，您的机器应该支持一些旧版本的 PyTorch，它允许 CUDA 7.5 或最好是 8.0（在撰写此答案时，最新版本使用最低 CUDA 9.2）。

但是，PyTorch 也需要 cuDNN。因此，cuDNN 6.0 适用于 CUDA 7.5。但是 cuDNN 6.0 需要 3.0 的计算能力。因此，大多数情况下，PyTorch 无法在您的机器上运行。（感谢指出cuDNN部分Robert Crovella）

【讨论】：

pytorch depends on cudnn 需要 cc3.0 或更高版本的 GPU。