pytorch cuda上tensor的定義 以及減少cpu的操作詳解
更新時間：2020年06月23日 10:15:02 作者：枯葉蝶KYD
這篇文章主要介紹了pytorch cuda上tensor的定義 以及減少cpu的操作詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
cuda上tensor的定義
a = torch.ones(1000,1000,3).cuda()
某一gpu上定義
cuda1 = torch.device('cuda:1')
b = torch.randn((1000,1000,1000),device=cuda1)
刪除某一變量
del a
在cpu定義tensor然后轉到gpu
torch.zeros().cuda()
直接在gpu上定義，這樣就減少了cpu的損耗
torch.cuda.FloatTensor(batch_size, self.hidden_dim, self.height, self.width).fill_(0)
補充知識：pytorch cuda.FloatTensor->FloatTensor
錯誤類型：
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor）
定義殘差塊時定義在model的外面，在使用gpu進行訓練的時候，殘差塊的參數是torch.FloatTensor類型，
雖然使用了model.cuda(),但是只對model里面的參數在gpu部分，所以把殘差塊對應的操作都在model的__init__(），
重新定義，即可解決問題
以上這篇pytorch cuda上tensor的定義 以及減少cpu的操作詳解就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
Python .py生成.pyd文件并打包.exe 的注意事項說明
這篇文章主要介紹了Python .py生成.pyd文件并打包.exe 的注意事項說明，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
本文主要是解決python2.7在查詢mysql時產生中文亂碼的處理方法，非常的有用，親測可行，推薦給大家2014-11-11
這篇文章主要介紹了Python的字典,結合實例形式詳細分析了Python字典的概念、創(chuàng)建、格式化及常用操作方法與相關注意事項,需要的朋友可以參考下2021-10-10
django restframework序列化字段校驗規(guī)則
本文主要介紹了django restframework序列化字段校驗規(guī)則，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2022-05-05
這篇文章主要為大家介紹了Python queue模塊攻略全解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-12-12
在音頻處理領域,背景噪聲是一個常見的問題,為了提高音頻的質量,我們需要對音頻進行降噪處理,本文將介紹如何使用 Python 實現音頻降噪,文中通過代碼示例講解的非常詳細,需要的朋友可以參考下2024-11-11
這篇文章主要介紹了如何實現Python 最短匹配模式，文中講解非常細致，代碼幫助大家更好的理解和學習，感興趣的朋友可以了解下2020-07-07
本文主要介紹了Tornado Application的實現，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-05-05
python雙向鏈表和單鏈表類似，只不過是增加了一個指向前面一個元素的指針,下面的代碼實例了python雙向鏈表的方法2013-11-11
撲克牌是一種廣泛使用的娛樂工具,通常用于各種紙牌游戲,本文給大家介紹了如何使用Python創(chuàng)建一個撲克牌生成器,需要的朋友可以參考下2025-04-04