Expected all tensors to be on the same device, but found at least two devices

Expected all tensors to be on the same device, but found at least two devices,

原因是计算的过程中,两个不同类型的变量在一起进行运算,即一个变量存储在gpu中,一个变量存储在cpu中,两个变量的存储位置冲突,导致无法计算,把变量统一下就行

python 复制代码
    x = x + kp_x_ofst
    y = y + kp_y_ofst
    x = torch.reshape(x, [-1])  
    y = torch.reshape(y, [-1])
    
    x = x.to('cpu')
    y = y.to('cpu')
 
    x0 = torch.floor(x).to(torch.int32)
    x1 = x0 + 1
    y0 = torch.floor(y).to(torch.int32)
    y1 = y0 + 1
    # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
    x0 = torch.clamp(x0, zero, max_x)
    x1 = torch.clamp(x1, zero, max_x)
    y0 = torch.clamp(y0, zero, max_y)
    y1 = torch.clamp(y1, zero, max_y)
相关推荐
小天才才1 小时前
【大模型】解耦大语言模型中的记忆与推理能力
人工智能·深度学习·语言模型·自然语言处理
nbbsn2 小时前
第四十天打卡
python·深度学习·机器学习
简诚2 小时前
前馈神经网络
深度学习
Blossom.1182 小时前
基于区块链的供应链溯源系统:构建与实践
人工智能·python·深度学习·机器学习·计算机视觉·flask·区块链
音程3 小时前
简单介绍Genetic Algorithms(遗传算法,简称 GA)
人工智能·深度学习·机器学习
deephub4 小时前
提升长序列建模效率:Mamba+交叉注意力架构完整指南
人工智能·深度学习·时间序列·mamba·交叉注意力
神经星星4 小时前
入选 ICML 2025,清华/人大提出统一生物分子动力学模拟器 UniSim
人工智能·深度学习·机器学习
vlln4 小时前
【论文解读】Search-R1:通过 RL 让 LLM 学会使用搜索引擎
人工智能·深度学习·神经网络·搜索引擎·transformer
盼小辉丶6 小时前
TensorFlow深度学习实战(21)——Transformer架构详解与实现
深度学习·tensorflow·transformer
databook8 小时前
规则学习:让机器学习像人类一样思考的可解释之路
python·机器学习·scikit-learn