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

python沒(méi)有g(shù)pu,如何改用cpu跑代碼

 更新時(shí)間:2022年05月16日 14:38:40   作者:孩紙D  
這篇文章主要介紹了python沒(méi)有g(shù)pu,如何改用cpu跑代碼?今天小編就為大家分享一下解決方案。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

沒(méi)有g(shù)pu,改用cpu跑代碼

以為很麻煩的修改,很大動(dòng)干戈的改動(dòng),以及無(wú)止盡的報(bào)錯(cuò),結(jié)果,就改動(dòng)代碼中的device就OK了,驚不驚喜,意不意外!?

也就是說(shuō),

只需要將device設(shè)置為以下即可

device=torch.device("cuda:0" if torch.cuda.is_available() else "cpu"))

另外也不需要全如此設(shè)置,只需查看報(bào)錯(cuò)處調(diào)用device的來(lái)源,在修改torch.device()括號(hào)里面的內(nèi)容就可。

Tensorflow-gpu禁用gpu設(shè)置(cpu與gpu速度對(duì)比)

禁用gpu設(shè)置

# 在import tensorflow之前
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

cpu與gpu對(duì)比

顯卡:GTX 1066

cpu

cpu

gpu

gpu

簡(jiǎn)單測(cè)試:gpu比cpu快5秒

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

相關(guān)文章

最新評(píng)論