问题
Pytorch 支持哪些 GPU,信息位于何处?
背景
Pytorch + GPU 的几乎所有文章都是关于 NVIDIA 的。NVIDIA 是 Pytorch 唯一可以使用的 GPU 吗?如果没有,哪些 GPU 可用,我在哪里可以找到信息?
Pytorch 支持哪些 GPU,信息位于何处?
Pytorch + GPU 的几乎所有文章都是关于 NVIDIA 的。NVIDIA 是 Pytorch 唯一可以使用的 GPU 吗?如果没有,哪些 GPU 可用,我在哪里可以找到信息?
没错,您需要与 CUDA 8、9 或 10 兼容的 NVIDIA GPU:https ://pytorch.org/get-started/locally/
NVIDIA 在此处提供了兼容卡列表:https ://developer.nvidia.com/cuda-gpus#compute
或者,您可以在配备 GPU 的云实例上工作(或在没有 GPU 的情况下安装 pytorch)。