Pytorch硬件要求

时间:2019-11-02 14:49:42

标签: deep-learning gpu pytorch

最新的PyTorch版本要求的最低计算能力是什么?

我有Nvidia Geforce 820M,其计算能力为2.1。如何在mu GPU (if it doesn't support naturally)

上运行PyTorch模型

1 个答案:

答案 0 :(得分:2)

this page看,PyTorch(甚至是最旧的版本)从版本7.5开始支持CUDA。鉴于this page,CUDA 7.5要求最低的计算能力2.0。因此,从表面上看,您的机器应支持允许CUDA 7.5或最好为8.0的PyTorch的某些较早版本(撰写本文时,最新版本使用的最低CUDA 9.2)。

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