欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

將pytorch的網(wǎng)絡等轉移到cuda

 更新時間:2021年06月01日 11:07:59   作者:aleien1  
這篇文章主要介紹了將pytorch的網(wǎng)絡等轉移到cuda的實現(xiàn),具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教

神經(jīng)網(wǎng)絡一般用GPU來跑,我們的神經(jīng)網(wǎng)絡框架一般也都安裝的GPU版本,本文就簡單記錄一下GPU使用的編寫。

GPU的設置不在model,而是在Train的初始化上。

第一步是查看是否可以使用GPU

self.GPU_IN_USE = torch.cuda.is_available()

就是返回這個可不可以用GPU的函數(shù),當你的pytorch是cpu版本的時候,他就會返回False。

然后是:

self.device = torch.device('cuda' if self.GPU_IN_USE else 'cpu')

torch.device是代表將torch.tensor分配到哪個設備的函數(shù)

接著是,我看到了一篇文章,原來就是將網(wǎng)絡啊、數(shù)據(jù)啊、隨機種子啊、損失函數(shù)啊、等等等等直接轉移到CUDA上就好了!

于是下面就好理解多了:

轉移模型:

self.model = Net(num_channels=1, upscale_factor=self.upscale_factor, base_channel=64, num_residuals=4).to(self.device)

設置cuda的隨機種子:

torch.cuda.manual_seed(self.seed)

轉移損失函數(shù):

self.criterion.cuda()

轉移數(shù)據(jù):

data, target = data.to(self.device), target.to(self.device)

pytorch 網(wǎng)絡定義參數(shù)的后面無法加.cuda()

pytorch定義網(wǎng)絡__init__()的時候,參數(shù)不能加“cuda()", 不然參數(shù)不包含在state_dict()中,比如下面這種寫法是錯誤的

self.W1 = nn.Parameter(torch.FloatTensor(3,3), requires_grad=True).cuda()

應該去掉".cuda()"

self.W1 = nn.Parameter(torch.FloatTensor(3,3), requires_grad=True)

以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。

相關文章

  • matplotlib實戰(zhàn)之餅圖繪制詳解

    matplotlib實戰(zhàn)之餅圖繪制詳解

    餅圖,或稱餅狀圖,是一個劃分為幾個扇形的圓形統(tǒng)計圖表,這篇文章主要為大家詳細介紹了如何使用Matplotlib繪制餅圖,需要的小伙伴可以參考下
    2023-08-08
  • python生成png的方法

    python生成png的方法

    本文主要介紹了python生成png的方法,文中通過示例代碼介紹的非常詳細,具有一定的參考價值,感興趣的小伙伴們可以參考一下
    2021-10-10
  • Python實現(xiàn)刪除windows下的長路徑文件

    Python實現(xiàn)刪除windows下的長路徑文件

    這篇文章主要為大家詳細介紹一下如何利用Python語言實現(xiàn)刪除windows下的長路徑文件功能,文中的示例代碼講解詳細,具有一定參考借鑒價值,感興趣的可以了解一下
    2022-07-07
  • 基于python爬取梨視頻實現(xiàn)過程解析

    基于python爬取梨視頻實現(xiàn)過程解析

    這篇文章主要介紹了基于python爬取梨視頻實現(xiàn)過程解析,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
    2020-11-11
  • python使用mysql的兩種使用方式

    python使用mysql的兩種使用方式

    pymsql是Python中操作MySQL的模塊。這篇文章主要介紹了python使用mysql的兩種使用方式,需要的朋友可以參考下
    2018-03-03
  • Python中jieba庫的介紹與使用

    Python中jieba庫的介紹與使用

    使用jieba庫對一段文本進行詞頻的統(tǒng)計是一件非常有意思的事,我們只需要使用這第三方庫,就可以在不閱讀文本的情況下,得到該文本的高頻率詞匯,這篇文章主要介紹了Python中jieba庫的介紹與使用,需要的朋友可以參考下
    2023-03-03
  • 舉例詳解Python中yield生成器的用法

    舉例詳解Python中yield生成器的用法

    這篇文章主要介紹了舉例詳解Python中yield生成器的用法,包括其在多線程multiprocess下的使用示例,非常推薦!需要的朋友可以參考下
    2015-08-08
  • 對pycharm代碼整體左移和右移縮進快捷鍵的介紹

    對pycharm代碼整體左移和右移縮進快捷鍵的介紹

    今天小編就為大家分享一篇對pycharm代碼整體左移和右移縮進快捷鍵的介紹,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2018-07-07
  • python實現(xiàn)自動整理文件

    python實現(xiàn)自動整理文件

    這篇文章主要介紹了python實現(xiàn)自動整理文件,主要內容通過整理桌面雜亂無章都是文檔和資料了解用python如何批量將不同后綴的文件移動到同一文件夾,需要的朋友可以參考一下
    2022-04-04
  • python matplotlib繪圖,修改坐標軸刻度為文字的實例

    python matplotlib繪圖,修改坐標軸刻度為文字的實例

    今天小編就為大家分享一篇python matplotlib繪圖,修改坐標軸刻度為文字的實例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2018-05-05

最新評論