將pytorch的網(wǎng)絡(luò)等轉(zhuǎn)移到cuda

更新時間：2021年06月01日 11:07:59 作者：aleien1

這篇文章主要介紹了將pytorch的網(wǎng)絡(luò)等轉(zhuǎn)移到cuda的實現(xiàn)，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

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

GPU的設(shè)置不在model，而是在Train的初始化上。

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

self.GPU_IN_USE = torch.cuda.is_available()

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

然后是：

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

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

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

于是下面就好理解多了：

轉(zhuǎn)移模型：

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

設(shè)置cuda的隨機(jī)種子：

torch.cuda.manual_seed(self.seed)

轉(zhuǎn)移損失函數(shù)：

self.criterion.cuda()

轉(zhuǎn)移數(shù)據(jù)：

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

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

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

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

應(yīng)該去掉".cuda()"

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

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

您可能感興趣的文章:

相關(guān)文章

python 解壓pkl文件的方法
今天小編就為大家分享一篇python 解壓pkl文件的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-10-10
Python關(guān)于迭代器的使用
這篇文章主要介紹了Python關(guān)于迭代器的使用,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
2024-06-06
Python實現(xiàn)將MySQL數(shù)據(jù)庫查詢結(jié)果導(dǎo)出到Excel
在實際工作中,我們經(jīng)常需要將數(shù)據(jù)庫中的數(shù)據(jù)導(dǎo)出到Excel表格中進(jìn)行進(jìn)一步的分析和處理,Python中的pymysql和xlsxwriter庫提供了很好的解決方案,下面我們就來看看具體操作方法吧
2023-11-11
Django 路由系統(tǒng)URLconf的使用
這篇文章主要介紹了Django 路由系統(tǒng)URLconf的使用，詳細(xì)的介紹了什么是URLconf以及如何使用，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2018-10-10
在樹莓派2或樹莓派B+上安裝Python和OpenCV的教程
這篇文章主要介紹了在樹莓派2或樹莓派B+上安裝Python和OpenCV的教程,主要基于GTK庫，并以Python2.7和OpenCV 2.4.X版本的安裝作為示例，需要的朋友可以參考下
2015-03-03
Python爬蟲之網(wǎng)絡(luò)請求
這篇文章主要介紹了Python爬蟲之網(wǎng)絡(luò)請求，文章基于Python展開對網(wǎng)絡(luò)請求的相關(guān)介紹，需要的小伙伴可以參考一下
2022-04-04
python中sklearn的pipeline模塊實例詳解
這篇文章主要介紹了python中sklearn的pipeline模塊的相關(guān)知識，本文通過實例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下
2020-05-05
Python3.6簡單反射操作示例
這篇文章主要介紹了Python3.6簡單反射操作,結(jié)合實例形式分析了Python3反射的概念、原理、相關(guān)操作技巧與注意事項,需要的朋友可以參考下
2018-06-06
Python使用xlrd模塊操作Excel數(shù)據(jù)導(dǎo)入的方法
這篇文章主要介紹了Python使用xlrd模塊操作Excel數(shù)據(jù)導(dǎo)入的方法,涉及Python操作xlrd模塊的技巧,需要的朋友可以參考下
2015-05-05
Python正則表達(dá)式教程之三：貪婪/非貪婪特性
這篇文章主要介紹了Python正則表達(dá)式中貪婪/非貪婪特性的相關(guān)資料，文中通過示例代碼介紹的很詳細(xì)，對大家具有一定的參考價值，需要的朋友下面來一起看看吧。
2017-03-03