pytorch吧 关注:1,400贴子:2,530
  • 3回复贴,共1

求助大佬帮帮萌新!!

只看楼主收藏回复

萌新是用笔记本训练的,任务管理器显示有两个GPU,一个是Intel的Graphics(GPU0),一个是3060(GPU1)。
在用pytorch搭建的transformer训练一个模型时,训练的开始阶段显示GPU1会有很高的峰值,之后就变为0,这个正常吗?


IP属地:陕西来自Android客户端1楼2025-02-17 12:50回复
    使用nvidia smi查看显卡使用情况


    IP属地:北京来自Android客户端2楼2025-02-17 22:46
    收起回复

      扫二维码下载贴吧客户端

      下载贴吧APP
      看高清直播、视频!