Intel Extension for PyTorch详细安装教程

Intel Extension for PyTorch详细安装教程

引用

CSDN

1.

https://blog.csdn.net/m0_63007797/article/details/136606591

Intel Extension for PyTorch是Intel为加速PyTorch深度学习框架而开发的扩展库。本文将详细介绍其安装步骤，包括系统要求、硬件需求、软件需求、驱动程序安装、Intel oneAPI Base Toolkit安装以及最终的完整性测试。

一、安装所需系统要求

1.1 硬件需求

• Intel® Arc™ A-Series GPUs（独立显卡）

1.2 软件需求

二、准备

2.1 安装驱动程序

首先，确保系统是Windows 10或Windows 11版本，并且拥有Intel Arc系列的显卡（核显或独显）。

2.2 安装Intel® oneAPI Base Toolkit

Intel® oneAPI Base Toolkit 安装地址

注意：
建议使用单独的特定于组件的激活脚本来逐个激活下面列出的所需组件。请查看下面的完整性测试部分，以获得一个使用示例。

• Intel®oneAPI dpc++编译器(安装路径为DPCPPROOT占位符)
• 英特尔®oneAPI数学内核库(oneMKL)(占位符MKLROOT作为其安装路径)

call {DPCPPROOT}\env\vars.bat
call {MKLROOT}\env\vars.bat
python -c "import torch; import intel_extension_for_pytorch as ipex; print(torch.__version__); print(ipex.__version__); [print(f'[{i}]: {torch.xpu.get_device_properties(i)}') for i in range(torch.xpu.device_count())];"

这个需要在完成以下步骤后运行最后一行Python命令。

三、安装

预构建的wheel文件可用于Python 3.8, 3.9, 3.10, 3.11。

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

完整性测试

安装完成后，运行以下命令进行完整性测试：

call {DPCPPROOT}\env\vars.bat
call {MKLROOT}\env\vars.bat
python -c "import torch; import intel_extension_for_pytorch as ipex; print(torch.__version__); print(ipex.__version__); [print(f'[{i}]: {torch.xpu.get_device_properties(i)}') for i in range(torch.xpu.device_count())];"

如果遇到以下警告：

:: WARNING: Visual Studio was not found in the standard installation location: "C:\Program Files\Microsoft Visual Studio\<Year>\<Edition>" or "C:\Program Files (x86)\Microsoft Visual Studio\<Year>\<Edition>" Set the VS2017INSTALLDIR or VS2019INSTALLDIR or VS2022INSTALLDIR environment variable to point to your install location and try again.

这通常意味着没有安装Visual Studio或安装路径不在默认位置。解决方案是重新安装Visual Studio，并根据C盘空间情况选择安装路径。如果C盘内存充裕，建议直接安装到C盘；如果C盘空间不足，可以安装到其他盘。正常情况下，仅选择必要的组件即可，占用空间约为1.5G左右。