pytorch判斷是否cuda 判斷變量類型方式
更新時間：2020年06月23日 10:49:55 作者：ShellCollector
這篇文章主要介紹了pytorch判斷是否cuda 判斷變量類型方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
我就廢話不多說了，那就直接看代碼吧~
inputs = Variable(torch.randn(2,2)) inputs.is_cuda # will return false inputs = Variable(torch.randn(2,2).cuda()) inputs.is_cuda # returns true
判斷：
torch.is_tensor() #如果是pytorch的tensor類型返回true
torch.is_storage() # 如果是pytorch的storage類型返回ture
這里還有一個小技巧，如果需要判斷tensor是否為空，可以如下
>>> a=torch.Tensor() >>> len(a) 0 >>> len(a) is 0 True
設置：通過一些內(nèi)置函數(shù)，可以實現(xiàn)對tensor的精度, 類型，print打印參數(shù)等進行設置
torch.set_default_dtype(d) #對torch.tensor() 設置默認的浮點類型 torch.set_default_tensor_type() # 同上，對torch.tensor()設置默認的tensor類型 >>> torch.tensor([1.2, 3]).dtype # initial default for floating point is torch.float32 torch.float32 >>> torch.set_default_dtype(torch.float64) >>> torch.tensor([1.2, 3]).dtype # a new floating point tensor torch.float64 >>> torch.set_default_tensor_type(torch.DoubleTensor) >>> torch.tensor([1.2, 3]).dtype # a new floating point tensor torch.float64 torch.get_default_dtype() #獲得當前默認的浮點類型torch.dtype torch.set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None）#) ## 設置printing的打印參數(shù)
判斷變量類型：下面兩種方法都行
if isinstance(downsample, torch.nn.Module):
# if torch.type(downsample) != torch.IntTensor:
補充知識：pytorch：測試GPU是否可用
廢話不多說，看代碼吧~
import torch
flag = torch.cuda.is_available()
print(flag)
ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())
True cuda:0 GeForce GTX 1080 tensor([[0.9530, 0.4746, 0.9819], [0.7192, 0.9427, 0.6768], [0.8594, 0.9490, 0.6551]], device='cuda:0')
以上這篇pytorch判斷是否cuda 判斷變量類型方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
Python實現(xiàn)實時增量數(shù)據(jù)加載工具的解決方案
這篇文章主要分享結合單例模式實際應用案例：實現(xiàn)實時增量數(shù)據(jù)加載工具的解決方案。最關鍵的是實現(xiàn)一個可進行添加、修改、刪除等操作的增量ID記錄表。需要的可以參考一下2022-02-02
解決Python中l(wèi)ist里的中文輸出到html模板里的問題
今天小編就為大家分享一篇解決Python中l(wèi)ist里的中文輸出到html模板里的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12
Qt Quick QML-500行代碼實現(xiàn)合成大西瓜游戲
合成大西瓜游戲是前段時間比較火的小游戲，最近小編閑來無事，通過研究小球碰撞原理親自寫碰撞算法實現(xiàn)一個合成大西瓜游戲，下面小編把我的實現(xiàn)思路及核心代碼分析出來，供大家參考2021-05-05
python manage.py runserver流程解析
這篇文章主要介紹了python manage.py runserver流程解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-11-11
Pytorch之nn.Upsample()和nn.ConvTranspose2d()用法詳解
nn.Upsample和nn.ConvTranspose2d是PyTorch中用于上采樣的兩種主要方法,nn.Upsample通過不同的插值方法（如nearest、bilinear）執(zhí)行上采樣,沒有可學習的參數(shù),適合快速簡單的尺寸增加,而nn.ConvTranspose2d通過可學習的轉(zhuǎn)置卷積核進行上采樣2024-10-10
Python中getservbyport和getservbyname函數(shù)的用法大全
在Python的網(wǎng)絡編程中，getservbyport()函數(shù)和getservbyname()函數(shù)是socket模塊中的兩個函數(shù)，因此在使用這兩個函數(shù)時，需要導入socket模塊，這篇文章主要介紹了Python中getservbyport和getservbyname函數(shù)的用法,需要的朋友可以參考下2023-01-01
這篇文章主要介紹了Python中的sorted函數(shù)使用解析,sorted()函數(shù)可以對可迭代對象進行排序,并且可以人為指定排序的依據(jù)以及方式,本文提供了解決與部分實現(xiàn)代碼,需要的朋友可以參考下2023-10-10
時間戳是一個表示日期和時間的數(shù)值,通常以秒為單位,在Python中,獲取時間戳是常見的任務,用于記錄事件、計時操作、以及在各種應用中跟蹤時間,本文將介紹多種獲取時間戳的方法,包括標準庫和第三方庫的方式,并提供示例代碼以幫助你更好地理解2023-11-11
這篇文章主要為大家詳細介紹了python實現(xiàn)windows壁紙定期更換功能，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-01-01
django authenticate用戶身份認證的項目實踐
Django的contrib.auth模塊中的authenticate()函數(shù)用于對用戶的憑據(jù)進行身份驗證,本文就來介紹一下django authenticate用戶身份認證的使用,具有一定的參考價值,感興趣的可以了解一下2023-08-08