zoukankan      html  css  js  c++  java
  • 【E-14-Pytorch】RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat1' in call to _th_addmm

    一、出错误背景:

    Pytorch 中想使用 CUDA 对程序计算进行加速

    二、问题分析

    错误的意思object device 类型期望得到的是 cuda 类型,但是实际上的类型确实 cpu 类型,在调用二分类交叉熵损失进行前向计算的时候

    三、检查下面几点:

    1. 模型是否放到了CUDA model = model.to(device)  model = model.cuda(device)
    2. 输入数据是否放到了CUDA data = data.to(device)  data = data.cuda(device)
    3. 模型内部新建的张量是否放到了CUDA p = torch.tensor([1]).to(device)  p = torch.tensor([1]).cuda(device)
      一般情况下应该是忘记了第三点,而根据提示也可以知道,在进行二分类交叉熵损失进行前向计算的过程中,存在没有放到cuda上的张量,找到他,fix it !!!

    其中:device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

    四、解决方式

       

  • 相关阅读:
    zabbix:乱码问题
    zabbix--微信报警(未完成)
    ansible-playbook项目(4)
    ansible-playbook(3)
    备份和校验脚本-邮件通知
    rsync
    keepalived
    双机热备
    nginx负载均衡
    LNMP(5)
  • 原文地址:https://www.cnblogs.com/yifanrensheng/p/13865617.html
Copyright © 2011-2022 走看看