最新的PyTorch版本要求的最低计算能力是什么?
我有Nvidia Geforce 820M,其计算能力为2.1。如何在mu GPU (if it doesn't support naturally)
答案 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)