Win Arc 770配置Pytorch xpu
Intel ARC Graphics 显卡基于Pytorch xpu进行训练环境
背景:WIn电脑不是N卡,是ARC 770 ,记录配置xpu的环境搭建过程
环境搭建要求:
英特尔独立显卡的驱动程序
Intel® oneAPI Base Toolkit (下载地址:https://www.intel.cn/content/www/cn/zh/developer/tools/oneapi/base-toolkit-download.html)
3 安装anaconda:
https://www.anaconda.com/download
anaconda下载推荐清华镜像源:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
或者百度一般anaconda下载安装的配置的博客安装一遍。
4、环境配置
开始搭建环境
1、显卡驱动程序就电脑管家或者驱动精灵默认安装即可。
2、Intel® oneAPI Base Toolkit
2.1 下载安装包
一般离线安装或者在线安装都可,看个人网络,离线安装包太大了,下载速度慢
一般默认安装就可以;
备注:oneAPI Base Toolkit 2025.1.0 需要安装 visual studio 2022
安装完成过后:还需要再安装目录去运行 setvars.bat
备注:setvars.bat 只对当前 cmd 命令行窗口生效。
3:安装anaconda 傻瓜式操作
3.2 创建虚拟环境
conda create -n pytorch_arc python=3.10
3.3 激活虚拟环境
conda activate pytorch_arc
安装pytorch xpu 版本
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
3.4pycharm中配置
在pycharm配置对应的虚拟环境 file->setting->project->project interpreter 点击add interpreter ,选择 select existing ,选择你在conda配置 envs 下面的环境对应的 python.exe即可
备注:pycharm 版本 2024.3.4,其他版本类似
PyCharm 2024.3.4 (Professional Edition)
Build #PY-243.25659.43, built on February 27, 2025
授权给 Taylor Mcbride
订阅有效期至 2025年12月10日。
For educational use only.
Runtime version: 21.0.6+8-b631.39 amd64 (JCEF 122.1.9)
VM: OpenJDK 64-Bit Server VM by JetBrains s.r.o.
Toolkit: sun.awt.windows.WToolkit
Windows 11.0
GC: G1 Young Generation, G1 Concurrent GC, G1 Old Generation
Memory: 4096M
Cores: 16
Registry:
ide.experimental.ui=true
terminal.new.ui=true
Non-Bundled Plugins:
com.intellij.ml.llm (243.23654.270.9)
com.github.copilot (1.5.40-243)
四、环境配置
问题:
因为oneAPI Base Toolkit 设置环境变量在虚拟环境未被继承,每次激活虚拟环境变量过后还得去执行一吧 setvars.bat
call "C:\Program Files (x86)\Intel\oneAPI\setvars.bat"
才能验证xpu成功
解决:
conda activate 自动设置环境量
但是每次都去执行一把这个 setvars.bat太麻烦了,如果我要用pycharm跑项目是不是也得去执行一把。还好我们在执行这个
conda activate 时回去调用我们本地bat我们去本地bat去给执行一把就行了,我的安装目录是D:\Develop\anaconda,pytorch_arc是我的虚拟环境名称,其他自己去拼接。
D:\Develop\anaconda\envs\pytorch_arc\etc\conda\activate.d
这个en_vars.bat是我直接创建的在里面执行你要运行和设置环境变量:
call "C:\Program Files (x86)\Intel\oneAPI\setvars.bat"
切换过后就去取设置环境变量,红框部分为自动执行
icx --version 验证是否加载oneAPI
或者 在python中
到这里就配置完成了