亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

為什么我們需要 image.to('CUDA') 當我們有 model.to('CUDA')

為什么我們需要 image.to('CUDA') 當我們有 model.to('CUDA')

揚帆大魚 2021-09-11 18:00:18
我正在上一門課程PyTorch。我想知道為什么我們需要單獨告訴torch.utils.data.DataLoader輸出它在什么設備上運行。如果模型已經打開,CUDA為什么它不相應地自動更改輸入?這種模式對我來說似乎很有趣:model.to(device)for ii, (inputs, labels) in enumerate(trainloader):    # Move input and label tensors to the GPU    inputs, labels = inputs.to(device), labels.to(device)是否有一個用例,我想讓模型在 GPU 上運行,但我的輸入是在 CPU 模式下,反之亦然?
查看完整描述

1 回答

  • 1 回答
  • 0 關注
  • 262 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號