如何使英特尔 GPU 可用于通过 pytorch 进行处理?

atp*_*atp 4 gpu deep-learning pytorch

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

Ale*_*yer 8

如果您使用的是英特尔最新的处理器之一,英特尔会为深度学习和机器学习提供优化的库。这篇文章是一个起点,它是关于 PyTorch 的英特尔优化入门的。他们在人工智能研讨会上提供了更多相关信息。


Rik*_*ika 5

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