Skip to content

https://blog.csdn.net/weixin_47345748/article/details/144192276 https://pytorch.ac.cn/docs/stable/notes/get_start_xpu.html

Intel ARC Graphics 显卡基于 Pytorch xpu 进行训练环境搭建指南

一、环境搭建要求

  1. 英特尔独立显卡的驱动程序：确保显卡驱动已安装。
  2. Intel® oneAPI Base Toolkit：下载地址：https://www.intel.cn/content/www/cn/zh/developer/tools/oneapi/base-toolkit-download.html
  3. 安装 Anaconda：推荐使用清华镜像源：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

二、开始搭建环境

1. 显卡驱动程序安装

使用电脑管家或者驱动精灵默认安装即可。

2. Intel® oneAPI Base Toolkit 安装

  • 可选择离线或在线安装包，楼主使用的是在线安装包。
  • 安装完成后，需要在安装目录下运行 setvars.bat。
  • 注意：setvars.bat 仅对当前 CMD 命令行窗口生效。

3. 安装 Anaconda 及配置虚拟环境

  • 安装 Anaconda 后，创建并激活虚拟环境：
conda create -n pytorch_arc python=3.12
conda activate pytorch_arc
  • 安装 Pytorch xpu 版本：
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
  • 在 Pycharm 中配置对应的虚拟环境。

三、环境配置及问题解决

1. 发现问题

运行时报错：“找不到指定的模块。Error loading ‘D:\ProgramData\anaconda3\envs\pytorch_arc\Lib\site-packages\torch\lib\c10_xpu.dll’ or one of its dependencies”

2. 手动解决

在激活虚拟环境后执行：

call "C:\Program Files (x86)\Intel\oneAPI\setvars.bat"

3. conda activate 自动设置环境变量

在虚拟环境的 activate.d 目录下创建 en_vars.bat 文件，内容为：

call "C:\Program Files (x86)\Intel\oneAPI\setvars.bat"

验证是否加载 oneAPI：

icx --version

或者在 Python 中：

import torch
torch.xpu.is_available()

四、参考