deep-learning - 如何使英特尔 GPU 可用于通过 pytorch 进行处理？

Question

我使用的是配备 Intel Corporation HD Graphics 520 的笔记本电脑。
有谁知道如何为深度学习设置它，特别是 Pytorch？我已经看到如果你有 Nvidia 显卡我可以安装 cuda 但是当你有英特尔 GPU 时该怎么办？

Answer 1

PyTorch 不支持 NVIDIA CUDA 和最近的 AMD Rocm 以外的任何东西。
英特尔在其他答案中对 Pytorch 的支持是至强系列处理器独有的，并且在 GPU 方面也没有那么可扩展。
英特尔 oneAPI以前的广告oneDNN然而，它支持广泛的硬件，包括英特尔的集成显卡，但目前，截至 2020 年 10 月 29 日或 PyTorch 1.7 尚未在 PyTorch 中实现全面支持。
但你还有其他选择。推断你有几个选择。
DirectML是其中之一。基本上你把你的模型转换成 onnx ，然后使用 directml provider 在 gpu 上运行您的模型(在我们的例子中将使用 DirectX12 并且目前仅适用于 Windows!)
您的另一个选择是使用 OpenVino和 TVM两者都支持多平台，包括Linux、Windows、Mac等。
它们都使用 ONNX 模型，因此您需要先将模型转换为 onnx 格式，然后再使用它们。