resnet50.to() 在非 NVIDIA GPU 上的函数
resnet50.to() function on a non-NVIDIA GPU
我正在尝试使用 Pytorch 函数将预训练的 ResNet50 模型转换为在 GPU 上使用 resnet50.to()。
问题是我在 Mac 上使用 Intel Iris Plus Graphics 655 1536 MB GPU，我不知道要传递给函数的参数是什么，因为我只找到了用于 NVIDIA GPU 的参数(resnet50.to('cuda:0'))。
PyTorch 使用 Nvidia 的 CUDA API 进行所有 GPU 交互。因此不支持其他不使用 CUDA API 的 GPU（例如 AMD 或 Intel GPU）。
如果您没有 Nvidia GPU，则无法在 GPU 上 运行 PyTorch。
我正在尝试使用 Pytorch 函数将预训练的 ResNet50 模型转换为在 GPU 上使用 resnet50.to()。
问题是我在 Mac 上使用 Intel Iris Plus Graphics 655 1536 MB GPU，我不知道要传递给函数的参数是什么，因为我只找到了用于 NVIDIA GPU 的参数(resnet50.to('cuda:0'))。
PyTorch 使用 Nvidia 的 CUDA API 进行所有 GPU 交互。因此不支持其他不使用 CUDA API 的 GPU（例如 AMD 或 Intel GPU）。
如果您没有 Nvidia GPU，则无法在 GPU 上 运行 PyTorch。