Intel® Extension for PyTorch*详细安装教程

最近在研究Intel的pytorch的加速拓展Intel® Extension for PyTorch*,但是发现官网的文档全是英文的,不太好找安装教程。所以特此分享Intel® Extension for PyTorch*的详细安装教程。

一、安装所需系统要求

1.1 硬件需求

  • Intel® Arc™ A-Series GPUs (独立显卡哦,集成显卡不行)

1.2 软件需求

在这里插入图片描述

二、准备

2.1 安装驱动程序

首先,我们的系统要是windows10或者windows 11版本,然后就是我们拥有英特尔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 或者 安装了路径不在默认位置(可能我们安装到了C盘以外地方)。

这里我是因为没有安装 Visual Studio ,所以我就重新安装了下。
Visual Studio 安装路径

在这里插入图片描述
这里如果C盘内存充裕的话,我建议直接放C盘就好,如果C盘不足,还是放到其他盘。
在这里插入图片描述
这里由于我还需要安装一个C++以及拓展,所以花费了20多G内存,正常情况下我们什么都不选用,只需要1.5G左右内存即可。

完整性测试

安装完Visual Studio以后,即可运行成功。
在这里插入图片描述

相关推荐

  1. npm详细安装教程

    2024-03-11 06:50:02       57 阅读
  2. RabbitMQ安装详细教程

    2024-03-11 06:50:02       36 阅读
  3. wifiphisher详细安装教程

    2024-03-11 06:50:02       27 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-03-11 06:50:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-03-11 06:50:02       100 阅读
  3. 在Django里面运行非项目文件

    2024-03-11 06:50:02       82 阅读
  4. Python语言-面向对象

    2024-03-11 06:50:02       91 阅读

热门阅读

  1. misc49

    misc49

    2024-03-11 06:50:02      44 阅读
  2. 从零开始学HCIA之IPv6基础05

    2024-03-11 06:50:02       43 阅读
  3. 2024年第十五届蓝桥杯第三期(校内)模拟赛题解

    2024-03-11 06:50:02       38 阅读
  4. 音视频之V4L2的应用

    2024-03-11 06:50:02       41 阅读
  5. VSCode 配置 Spring Boot 项目开发环境

    2024-03-11 06:50:02       45 阅读
  6. 算法 Tips

    2024-03-11 06:50:02       43 阅读
  7. 机器视觉学习(二)—— 显示图像和视频

    2024-03-11 06:50:02       41 阅读
  8. Jenkins Error 403 No valid crumb was included in the request

    2024-03-11 06:50:02       37 阅读