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

Question

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

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

标签： pythondeep-learningpytorchgpuresnet

Answer 1

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

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