deep-learning - 如何通过 pytorch 使 Intel GPU 可用于处理？

Question

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

标签： deep-learningpytorchgpu

Answer 1

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