非 NVIDIA GPU 上的 resnet50.to() 函数

Question

我正在尝试使用 Pytorch 函数 resnet50.to() 将预训练的 ResNet50 模型转换为在 GPU 上使用。

问题是我在 Mac 上使用的是 Intel Iris Plus Graphics 655 1536 MB GPU，我不知道传递给函数的参数是什么，因为我只找到了一个用于 NVIDIA GPU (resnet50.to('cuda:0')) 的参数。

Answer 1

PyTorch 使用 Nvidia 的 CUDA API 进行所有 GPU 交互。因此不支持其他不使用 CUDA API 的 GPU（例如 AMD 或 Intel GPU）。

如果您没有 Nvidia GPU，则无法在 GPU 上运行 PyTorch。