如何让Intel GPU可以通过pytorch进行处理?

问题描述 投票:0回答:3

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

deep-learning pytorch gpu intel
3个回答
16
投票

PyTorch 不支持 NVIDIA CUDA 和最近的 AMD Rocm 以外的任何东西。 其他答案中给出的英特尔对 Pytorch 的支持是 Xeon 系列处理器独有的,并且在 GPU 方面也没有那么可扩展。
Intel 的

oneAPI
(以前称为广告
oneDNN
)支持多种硬件,包括 Intel 的集成显卡,但目前,截至 2020 年 10 月 29 日或 PyTorch 1.7 尚未在 PyTorch 中实现全面支持。
但你还有其他选择。作为推论,你有几个选择。 DirectML 就是其中之一。基本上,您将模型转换为 onnx,然后使用 directml 提供程序在 GPU 上运行您的模型(在我们的例子中将使用 DirectX12,目前仅适用于 Windows!)

您的另一个选择是使用OpenVinoTVM,它们都支持多平台,包括Linux、Windows、Mac等。
OpenVino 和 TVM 使用 ONNX 模型,因此您需要首先将模型转换为 onnx 格式,然后再使用它们。

最近(截至 2023 年),IREE(中间表示执行环境)(在本例中为torch-mlir)也可以使用。


8
投票

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


0
投票

PyTorch v2.4 开始支持 Intel GPU。请检查这些链接以获取更多信息:

  • https://pytorch.org/docs/stable/notes/get_start_xpu.html
  • https://github.com/intel/intel-extension-for-pytorch
© www.soinside.com 2019 - 2024. All rights reserved.