intel/pytorch

Intel®硬件上的PyTorch*镜像

镜像概述

本镜像由Intel官方提供，基于Ubuntu 24.04系统构建，针对Intel CPU和GPU进行性能优化，集成PyTorch框架，帮助开发者快速搭建适用于Intel硬件的深度学习开发环境。

核心功能

Intel® CPU支持

PyTorch在Intel CPU上利用以下指令集提升计算效率：

• Intel®高级矩阵扩展（Intel® AMX）
• Intel®高级向量扩展512（Intel® AVX-512）
• 向量神经网络指令（VNNI）

Intel® GPU支持

通过xpu设备后端，支持以下Intel GPU型号：

• Intel® Arc A系列显卡
• Intel® Arc B系列显卡
• 搭载Intel® Arc™显卡的Intel® Core™ Ultra处理器（系列2、3）
• 搭载Intel® Arc™显卡的Intel® Core™ Ultra移动处理器（系列2、3）

使用场景

• 深度学习模型的训练与推理任务
• 在Intel CPU/GPU上优化运行PyTorch应用
• 快速验证Intel硬件对深度学习任务的加速效果

配置说明

CPU版本标签及部署

CPU版本运行示例：

bash
docker run -it --rm \
    --ipc=host \
    docker.xuanyuan.run/intel/pytorch:cpu-2.10.0

GPU版本标签及部署

GPU版本运行示例：

bash
DEVICE=${DEVICE:-/dev/dri/renderD128}
DEVICE_GRP=$(stat --format %g $DEVICE)
docker run -it --rm \
    --device $DEVICE \
    --group-add $DEVICE_GRP \
    -v /dev/dri/by-path:/dev/dri/by-path \
    --ipc=host \
    intel/pytorch:latest

已知限制

以下Intel GPU未完全验证，不保证兼容：

• Intel®数据中心GPU Max系列
• Intel®数据中心GPU Flex系列

附加说明

• 镜像Dockerfile可在https://github.com/intel/containers%E4%BB%93%E5%BA%93%E8%8E%B7%E5%8F%96
• 使用需遵守https://github.com/intel/containers/tree/main/LICENSE
• Ubuntu基础镜像仅用于演示，生产环境建议启用Ubuntu Pro以获得扩展安全维护
• 所有Intel商标及第三方商标归各自所有者所有