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用pdfplumber提取pdf表格數據并保存到excel文件中
在實際研究中我們經常需要獲取大量數據,而這些數據很大一部分以pdf表格的形式呈現,如公司年報、發(fā)行上市公告等,下面這篇文章主要給大家介紹了關于利用python提取pdf表格數據并保存到excel文件中的相關資料,需要的朋友可以參考下2022-07-07
今天小編就為大家分享一篇使用Python 統計高頻字數的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
這篇文章主要介紹了python 使用OpenCV進行簡單的人像分割與合成的方法，幫助大家更好的利用python處理圖像，感興趣的朋友可以了解下2021-02-02
這篇文章主要介紹了python按行讀取文件并找出其中指定字符串的方法，本文通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了如何用python 批量重命名移動多個文件，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下2021-03-03
這篇文章主要介紹了利用Python對中國500強排行榜數據進行可視化分析，從不同角度去對數據進行統計分析，可視化展示，下文詳細內容介紹需要的小伙伴可以參考一下2022-05-05
這篇文章主要介紹了解決TensorFlow訓練模型及保存數量限制的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
python Matplotlib數據可視化（2）：詳解三大容器對象與常用設置
這篇文章主要介紹了python Matplotlib三大容器對象與常用設置的相關資料，幫助大家更好的學習和使用Matplotlib庫的用法，感興趣的朋友可以了解下2020-09-09
通過Python實現將你的 QQ 頭像（或者微博頭像）右上角加上紅色的數字，類似于微信未讀信息數量那種提示效果。下面通過本文給大家分享python實現微信頭像加紅色數字功能，感興趣的朋友一起看看吧2018-03-03
這篇文章主要介紹了使用Python的Matplotlib繪制漸變的彩色曲線,Matplotlib是Python生態(tài)圈中應用最為廣泛的2D繪圖庫,通過擴展工具包mplot3d,也可以繪制簡單的3D模型,需要的朋友可以參考下2023-08-08