Pytorch 支持的 GPU

数据挖掘 火炬 显卡
2021-09-16 05:57:27

问题

Pytorch 支持哪些 GPU,信息位于何处?

背景

Pytorch + GPU 的几乎所有文章都是关于 NVIDIA 的。NVIDIA 是 Pytorch 唯一可以使用的 GPU 吗?如果没有,哪些 GPU 可用,我在哪里可以找到信息?

1个回答

没错,您需要与 CUDA 8、9 或 10 兼容的 NVIDIA GPU:https ://pytorch.org/get-started/locally/

NVIDIA 在此处提供了兼容卡列表:https ://developer.nvidia.com/cuda-gpus#compute

或者,您可以在配备 GPU 的云实例上工作(或在没有 GPU 的情况下安装 pytorch)。