Data Science Stack Exchange is a question and answer site for Data science professionals, Machine Learning specialists, and those interested in learning more about the field. It only takes a minute to sign up.
Sign up to join this community
Which GPUs are supported in Pytorch and where is the information located?
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?
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).
