Intel XE核显跑pytorch踩坑记录_intel庐 extension for pytorch-CSDN博客

本文链接：https://blog.csdn.net/qq_37160051/article/details/147146532

给想用核显跑pytorch的同学一点参考

一、安装需求

intel 11代及以后的CPU
windows10及以后的操作系统或Linux系统

二、准备工作

更新核显驱动到最新版
安装Intel® oneAPI Base Toolkit
安装python（注意版本要求，这里推荐使用python3.10）

三、安装

按如下命令安装：

conda install pkg-config libuv
python -m pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 intel-extension-for-pytorch==2.5.10 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/

注意了，是 intel-extension-for-pytorch==2.5.10，不是 intel-extension-for-pytorch==2.5.1，也不是 intel_extension_for_pytorch==2.5.10，坑死了。

四、验证

进入python，运行如下代码进行验证：

import torch
import intel_extension_for_pytorch as ipex

print(torch.__version__)
print(ipex.__version__)

# 检查 XPU 是否可用
# if torch.xpu.is_available():
if ipex.is_available():
    device = torch.device("xpu")
    print(f"XPU 设备名称: {torch.xpu.get_device_name(0)}")  # 应显示 Intel(R) Iris Xe Graphics
else: print("XPU 不可用，请检查驱动和安装！")

没报什么错的话基本就成功了，祝好运！

常见问题

您可能会遇到以下问题：

OSError: [WinError 126] 找不到指定的模块。 Error loading "C:\ProgramData\miniforge3\envs\xpu\lib\site-packages\torch\lib\backend_with_compiler.dll" or one of its dependencies.

解决方法：
①上面说的oneAPI和pkg-config libuv记得装欸！！！
②可以尝试安装Visual C++ Redistributable或安装Visual Studio
③尝试pip install mkl-dpcpp dpcpp-cpp-rt

还是没解决的话，自己去这里慢慢逛吧。