TransWikia.com

Supported GPU for Pytorch

Data Science Asked on January 31, 2021

Question

Which GPUs are supported in Pytorch and where is the information located?

Background

Almost all articles of Pytorch + GPU are about NVIDIA. Is NVIDIA the only GPU that can be used by Pytorch? If not, which GPUs are usable and where I can find the information?

One Answer

That's correct, you need a NVIDIA GPU compatible with CUDA 8, 9 or 10: https://pytorch.org/get-started/locally/

NVIDIA has a list of compatible cards here: https://developer.nvidia.com/cuda-gpus#compute

Alternatively you could work on a GPU equipped cloud instance (or install pytorch without GPU).

Answered by redhqs on January 31, 2021

Add your own answers!

Ask a Question

Get help from others!

© 2024 TransWikia.com. All rights reserved. Sites we Love: PCI Database, UKBizDB, Menu Kuliner, Sharing RPP