Pytorch Требования к оборудованию - PullRequest
0 голосов
/ 02 ноября 2019

Какова минимальная вычислительная мощность, необходимая для последней версии PyTorch?

У меня есть Nvidia Geforce 820M с возможностью вычислений 2.1. Как запустить модели PyTorch на графическом процессоре mu (if it doesn't support naturally)

1 Ответ

2 голосов
/ 02 ноября 2019

Глядя на эту страницу , PyTorch (даже несколько самых старых версий) поддерживает CUDA и выше от версии 7.5. Принимая во внимание, что, глядя на эту страницу , 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 Роберт Кровелла )

...