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

如何判斷pytorch是否支持GPU加速

 更新時(shí)間:2021年06月01日 09:40:34   作者:抬頭仰望-y  
這篇文章主要介紹了判斷pytorch是否支持GPU加速的操作,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

加上這句代碼:

print torch.cuda.is_available()

判斷完畢!說(shuō)說(shuō)在pytorch中如何查看gpu信息吧~

為什么將數(shù)據(jù)轉(zhuǎn)移至GPU的方法叫做.cuda而不是.gpu,就像將數(shù)據(jù)轉(zhuǎn)移至CPU調(diào)用的方法是.cpu?這是因?yàn)镚PU的編程接口采用CUDA,而目前并不是所有的GPU都支持CUDA,只有部分Nvidia的GPU才支持。

PyTorch未來(lái)可能會(huì)支持AMD的GPU,而AMD GPU的編程接口采用OpenCL,因此PyTorch還預(yù)留著.cl方法,用于以后支持AMD等的GPU。

torch.cuda.is_available()

cuda是否可用;

torch.cuda.device_count()

返回gpu數(shù)量;

torch.cuda.get_device_name(0)

返回gpu名字,設(shè)備索引默認(rèn)從0開始;

torch.cuda.current_device()

返回當(dāng)前設(shè)備索引;

在這里插入圖片描述

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

相關(guān)文章

最新評(píng)論