为PyTorch启用英特尔GPU支持 | AI生成和翻译
进展顺利！PyTorch 和 IPEX 现已协同工作。不过您的系统尚未启用英特尔 GPU 支持。让我帮您了解当前状况及解决方法：
当前状态
- ✓ PyTorch + IPEX 安装正确
- ✗ 英特尔 GPU（XPU）支持不可用
- ✗ 缺少可选软件包（
dpctl、scikit-learn-intelex）
启用英特尔 GPU 支持
1. 检查是否配备英特尔 GPU
# 检查 GPU 硬件
lspci | grep -i vga
# 或
lspci | grep -i display
2. 安装英特尔 GPU 驱动程序（若使用英特尔锐炫/锐炬/UHD 显卡）
Ubuntu/Debian 系统：
# 添加英特尔 GPU 软件源
wget -qO - https://repositories.intel.com/gpu/intel-graphics.key | \
sudo gpg --dearmor --output /usr/share/keyrings/intel-graphics.gpg
echo "deb [arch=amd64 signed-by=/usr/share/keyrings/intel-graphics.gpg] https://repositories.intel.com/gpu/ubuntu jammy client" | \
sudo tee /etc/apt/sources.list.d/intel-gpu-jammy.list
sudo apt update
# 安装 Level Zero 和 OpenCL 运行时
sudo apt install -y \
intel-opencl-icd \
intel-level-zero-gpu \
level-zero \
level-zero-dev
3. 安装附加 Python 软件包
# 安装用于英特尔 GPU 设备管理的 dpctl
pip install dpctl
# 安装英特尔优化的 scikit-learn
pip install scikit-learn-intelex
4. 安装支持 XPU 的 PyTorch
# 卸载仅支持 CPU 的 PyTorch
pip uninstall torch torchvision torchaudio
# 安装支持英特尔 XPU 的 PyTorch
pip install torch==2.8.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/xpu
# 确保 IPEX 版本匹配
pip install intel-extension-for-pytorch==2.8.0
5. 验证安装
python -c "import torch; import intel_extension_for_pytorch as ipex; print('XPU 可用:', torch.xpu.is_available()); print('XPU 设备数量:', torch.xpu.device_count() if torch.xpu.is_available() else 0)"
关于训练错误
错误提示 "The optimizer should be given for training mode" 是脚本中的缺陷。在训练时需要传入优化器参数。请检查 try_intel_gpu.py 文件，确保训练函数包含优化器参数。
您使用的是哪种英特尔 GPU 硬件？ 请运行 lspci | grep -i vga 命令查看，我将为您提供更具体的指导。