0
  • 聊天消息
  • 系統(tǒng)消息
  • 評論與回復
登錄后你可以
  • 下載海量資料
  • 學習在線課程
  • 觀看技術視頻
  • 寫文章/發(fā)帖/加入社區(qū)
會員中心
創(chuàng)作中心

完善資料讓更多小伙伴認識你,還能領取20積分哦,立即完善>

3天內(nèi)不再提示

開發(fā)者的福音:PyTorch 2.5現(xiàn)已支持英特爾獨立顯卡訓練

jf_23871869 ? 來源:jf_23871869 ? 作者:jf_23871869 ? 2024-10-30 13:48 ? 次閱讀
加入交流群
微信小助手二維碼

掃碼添加小助手

加入工程師交流群

[《PyTorch 2.5重磅更新:性能優(yōu)化+新特性》]中的一個新特性就是:正式支持在英特爾^?^獨立顯卡上訓練模型!

PyTorch 2.5
獨立顯卡類型
Intel^?^數(shù)據(jù)中心GPU Max系列
Intel^?^Arc?系列

本文將在Intel^?^Core? Ultra 7 155H自帶的Arc?集成顯卡上展示使用Pytorch2.5搭建并訓練AI模型的全流程。

,搭建開發(fā)環(huán)境

首先,請安裝顯卡驅(qū)動,參考指南:

https://dgpu-docs.intel.com/driver/client/overview.html

然后,**請下載并安裝Anaconda,**鏈接↓↓

https://www.anaconda.com/download

并用下面的命令創(chuàng)建并激活名為pytorch_arc的虛擬環(huán)境:

conda create -n pytorch_arc python=3.11    #創(chuàng)建虛擬環(huán)境
conda activate pytorch_arc                 #激活虛擬環(huán)境
python -m pip install --upgrade pip        #升級pip到最新版本

接著,安裝Pytorch XPU版;

pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu

滑動查看更多

最后,執(zhí)行命令,驗證安裝??吹椒祷亟Y(jié)果為“True”,證明環(huán)境搭建成功!

>> > import torch
 >> > torch.xpu.is_available()

訓練ResNet模型

執(zhí)行下載的訓練代碼,實現(xiàn)在Intel^?^ Arc?集成顯卡上訓練ResNet50模型。代碼下載鏈接:

https://gitee.com/Pauntech/Pytorch-2.5
import torch
import torchvision


LR = 0.001
DOWNLOAD = True
DATA = "datasets/cifar10/"


transform = torchvision.transforms.Compose(
    [
        torchvision.transforms.Resize((224, 224)),
        torchvision.transforms.ToTensor(),
        torchvision.transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
    ]
)
train_dataset = torchvision.datasets.CIFAR10(
    root=DATA,
    train=True,
    transform=transform,
    download=DOWNLOAD,
)
train_loader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=128)
train_len = len(train_loader)


model = torchvision.models.resnet50()
criterion = torch.nn.CrossEntropyLoss()
optimizer = torch.optim.SGD(model.parameters(), lr=LR, momentum=0.9)
model.train()
model = model.to("xpu")
criterion = criterion.to("xpu")


print(f"Initiating training")
for batch_idx, (data, target) in enumerate(train_loader):
    data = data.to("xpu")
    target = target.to("xpu")
    optimizer.zero_grad()
    output = model(data)
    loss = criterion(output, target)
    loss.backward()
    optimizer.step()
    if (batch_idx + 1) % 10 == 0:
         iteration_loss = loss.item()
         print(f"Iteration [{batch_idx+1}/{train_len}], Loss: {iteration_loss:.4f}")
torch.save(
    {
        "model_state_dict": model.state_dict(),
        "optimizer_state_dict": optimizer.state_dict(),
    },
    "checkpoint.pth",
)


print("Execution finished")

,總結(jié)

使用[PyTorch]在英特爾獨立顯卡上訓練模型將為AI行業(yè)新增計算硬件選擇!

審核編輯 黃宇

聲明:本文內(nèi)容及配圖由入駐作者撰寫或者入駐合作網(wǎng)站授權轉(zhuǎn)載。文章觀點僅代表作者本人,不代表電子發(fā)燒友網(wǎng)立場。文章及其配圖僅供工程師學習之用,如有內(nèi)容侵權或者其他違規(guī)問題,請聯(lián)系本站處理。 舉報投訴
  • 英特爾
    +關注

    關注

    61

    文章

    10187

    瀏覽量

    174253
  • 顯卡
    +關注

    關注

    16

    文章

    2505

    瀏覽量

    69320
  • pytorch
    +關注

    關注

    2

    文章

    809

    瀏覽量

    13849
收藏 0人收藏
加入交流群
微信小助手二維碼

掃碼添加小助手

加入工程師交流群

    評論

    相關推薦
    熱點推薦

    英特爾發(fā)布全新GPU,AI和工作站迎來新選擇

    英特爾推出面向準專業(yè)用戶和AI開發(fā)者英特爾銳炫Pro GPU系列,發(fā)布英特爾? Gaudi 3 AI加速器機架級和PCIe部署方案 ? 2025 年 5 月 19 日,北京 ——今日
    發(fā)表于 05-20 11:03 ?1483次閱讀

    2025英特爾人工智能創(chuàng)新應用大賽正式啟動

    近日,2025英特爾人工智能創(chuàng)新應用大賽(以下簡稱“大賽”)正式啟動。本屆大賽以“‘碼’上出發(fā),‘芯’創(chuàng)未來”為主題,在賽制、規(guī)模、獎項和賽事支持上實現(xiàn)多重升級,為開發(fā)者和企業(yè)提供展示創(chuàng)意和成果的廣闊平臺,鼓勵他們充分利用
    的頭像 發(fā)表于 04-02 15:24 ?497次閱讀

    英特爾銳炫B系列顯卡XeSS 2技術正式支持網(wǎng)易游戲

    的合作正在重新定義玩家的視覺體驗。 首款支持XeSS 2的國產(chǎn)大作“芯”動來襲 近日,英特爾銳炫B系列顯卡XeSS 2技術正式支持由網(wǎng)易游戲研發(fā)的《漫威爭鋒》,使其成為中國首款
    的頭像 發(fā)表于 03-08 09:19 ?676次閱讀

    英特爾?獨立顯卡與OpenVINO?工具套件結(jié)合使用時,無法運行推理怎么解決?

    使用英特爾?獨立顯卡與OpenVINO?工具套件時無法運行推理
    發(fā)表于 03-05 06:56

    探索座艙“未至之境”:英特爾支持DeepSeek車內(nèi)本地運行

    車載獨立顯卡的發(fā)布,更是將能支持的模型參數(shù)推至32B。當下經(jīng)典的AI駕艙應用,比如,駕駛員與AI助手討論明天的行程,關于沿途各種地標的語音交互,乃至于如語音調(diào)節(jié)溫度、問詢操作指引等經(jīng)典應用場景,由于DeepSeek的到來,會變得
    的頭像 發(fā)表于 02-14 16:19 ?336次閱讀
    探索座艙“未至之境”:<b class='flag-5'>英特爾</b><b class='flag-5'>支持</b>DeepSeek車內(nèi)本地運行

    英特爾CEO霍爾索斯于CES 2025重申獨立顯卡市場戰(zhàn)略

    近日,在CES 2025的主題演講中,英特爾新任聯(lián)席首席執(zhí)行官米歇爾-約翰斯頓-霍爾索斯向與會強調(diào):“獨立顯卡市場對我們至關重要,我們將持續(xù)在該領域進行戰(zhàn)略投資?!边@一表態(tài)無疑是對外
    的頭像 發(fā)表于 01-08 14:31 ?556次閱讀

    使用英特爾AI PC為YOLO模型訓練加速

    之后,情況有了新的變化,PyTorch2.5正式開始支持英特爾顯卡,也就是說,此后我們能夠借助英特爾 銳炫
    的頭像 發(fā)表于 12-09 16:14 ?1525次閱讀
    使用<b class='flag-5'>英特爾</b>AI PC為YOLO模型<b class='flag-5'>訓練</b>加速

    英特爾推出全新英特爾銳炫B系列顯卡

    英特爾銳炫B580和B570 GPU以卓越價值為時新游戲帶來超凡表現(xiàn)。 ? > 今日,英特爾發(fā)布全新英特爾銳炫 B系列顯卡(代號Battlemage)。
    的頭像 發(fā)表于 12-07 10:16 ?1391次閱讀
    <b class='flag-5'>英特爾</b>推出全新<b class='flag-5'>英特爾</b>銳炫B系列<b class='flag-5'>顯卡</b>

    英特爾聯(lián)合中科創(chuàng)達構建下一代智能座艙平臺

    近日,英特爾 AI 座艙暨車載獨立顯卡發(fā)布會在深圳盛大舉行。英特爾震撼發(fā)布其首款車載獨立顯卡 d
    的頭像 發(fā)表于 11-17 11:11 ?1024次閱讀

    使用PyTorch英特爾獨立顯卡訓練模型

    PyTorch 2.5重磅更新:性能優(yōu)化+新特性》中的一個新特性就是:正式支持英特爾獨立顯卡
    的頭像 發(fā)表于 11-01 14:21 ?1972次閱讀
    使用<b class='flag-5'>PyTorch</b>在<b class='flag-5'>英特爾</b><b class='flag-5'>獨立</b><b class='flag-5'>顯卡</b>上<b class='flag-5'>訓練</b>模型

    英特爾聚焦AI座艙

    英特爾推出首款銳炫車載獨立顯卡(dGPU)和第一代英特爾軟件定義車載SoC系列,滿足當前消費對汽車內(nèi)部配備更多屏幕、獲得更高清晰度等AI座
    的頭像 發(fā)表于 10-30 16:26 ?421次閱讀

    支持140億參數(shù)AI模型,229TOPS!英特爾重磅發(fā)布第一代車載獨立顯卡

    英特爾院士、英特爾公司副總裁、汽車事業(yè)部總經(jīng)理Jack Weast指出,在今年的CES上,英特爾發(fā)布了第一代AI增強型軟件定義車載SoC。8月8日,英特爾正式推出首款
    的頭像 發(fā)表于 08-12 09:07 ?1.3w次閱讀
    <b class='flag-5'>支持</b>140億參數(shù)AI模型,229TOPS!<b class='flag-5'>英特爾</b>重磅發(fā)布第一代車載<b class='flag-5'>獨立</b><b class='flag-5'>顯卡</b>

    英特爾發(fā)布第一代車載銳炫獨立顯卡

    英特爾震撼發(fā)布其第一代車載英特爾銳炫獨立顯卡,標志著智能座艙技術邁入全新階段。這款顯卡平臺算力高達229TOPS,不僅
    的頭像 發(fā)表于 08-09 14:54 ?802次閱讀

    英特爾正式推出第一代車載獨立顯卡

    8月8日,英特爾公司正式推出首款英特爾銳炫?車載獨立顯卡(dGPU),以重塑汽車行業(yè)格局。這一全新產(chǎn)品將賦能汽車廠商打造下一代車載體驗,以滿足并超越當前消費
    的頭像 發(fā)表于 08-09 09:27 ?8508次閱讀
    <b class='flag-5'>英特爾</b>正式推出第一代車載<b class='flag-5'>獨立</b><b class='flag-5'>顯卡</b>

    從運動員到開發(fā)者: 英特爾以開放式AI系統(tǒng)應對多重挑戰(zhàn)

    打造的生成式AI(GenAI)檢索增強生成(RAG)解決方案。該成果深度展示了英特爾如何通過基于英特爾?至強?處理器和英特爾??Gaudi AI加速器的開放式AI系統(tǒng)及平臺,幫助開發(fā)者
    的頭像 發(fā)表于 07-25 09:28 ?488次閱讀
    從運動員到<b class='flag-5'>開發(fā)者</b>: <b class='flag-5'>英特爾</b>以開放式AI系統(tǒng)應對多重挑戰(zhàn)

    電子發(fā)燒友

    中國電子工程師最喜歡的網(wǎng)站

    • 2931785位工程師會員交流學習
    • 獲取您個性化的科技前沿技術信息
    • 參加活動獲取豐厚的禮品