快捷導(dǎo)航

詳解tf.device()指定tensorflow運(yùn)行的GPU或CPU設(shè)備實(shí)現(xiàn)

更新時(shí)間：2021年02月20日 11:12:19 作者：-牧野-

這篇文章主要介紹了詳解tf.device()指定tensorflow運(yùn)行的GPU或CPU設(shè)備實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

在tensorflow中，我們可以使用 tf.device() 指定模型運(yùn)行的具體設(shè)備，可以指定運(yùn)行在GPU還是CUP上，以及哪塊GPU上。

設(shè)置使用GPU

使用 tf.device('/gpu:1') 指定Session在第二塊GPU上運(yùn)行：

import tensorflow as tf
 
with tf.device('/gpu:1'):
  v1 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v1')
  v2 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v2')
  sumV12 = v1 + v2
 
  with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:
    print sess.run(sumV12)

ConfigProto() 中參數(shù) log_device_placement=True 會(huì)打印出執(zhí)行操作所用的設(shè)備，以上輸出：

如果安裝的是GPU版本的tensorflow，機(jī)器上有支持的GPU，也正確安裝了顯卡驅(qū)動(dòng)、CUDA和cuDNN，默認(rèn)情況下，Session會(huì)在GPU上運(yùn)行：

import tensorflow as tf
 
v1 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v1')
v2 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v2')
sumV12 = v1 + v2
 
with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:
  print sess.run(sumV12)

默認(rèn)在GPU:0上執(zhí)行：

設(shè)置使用cpu

tensorflow中不同的GPU使用/gpu:0和/gpu:1區(qū)分，而CPU不區(qū)分設(shè)備號(hào)，統(tǒng)一使用 /cpu:0

import tensorflow as tf
 
with tf.device('/cpu:0'):
  v1 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v1')
  v2 = tf.constant([1.0, 2.0, 3.0], shape=[3], name='v2')
  sumV12 = v1 + v2
 
  with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:
    print sess.run(sumV12)

到此這篇關(guān)于詳解tf.device()指定tensorflow運(yùn)行的GPU或CPU設(shè)備實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)tensorflow運(yùn)行GPU或CPU內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

相關(guān)文章

anaconda打開(kāi)閃退的解決過(guò)程
這篇文章主要給大家介紹了關(guān)于anaconda打開(kāi)閃退的解決過(guò)程,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),對(duì)大家學(xué)習(xí)或者使用anaconda具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2022-09-09
python中hashlib模塊用法示例
這篇文章主要介紹了python中hashlib模塊用法示例，具有一定參考價(jià)值，需要的朋友可以了解下。
2017-10-10
python中@符號(hào)實(shí)例詳解
@是一個(gè)裝飾器,針對(duì)函數(shù),起調(diào)用傳參的作用,下面這篇文章主要給大家介紹了關(guān)于python中@符號(hào)的相關(guān)資料,文中通過(guò)示例代碼介紹的非常詳細(xì),需要的朋友可以參考下
2022-12-12
python執(zhí)行g(shù)et提交的方法
這篇文章主要介紹了python執(zhí)行g(shù)et提交的方法,實(shí)例分析了Python通過(guò)get傳遞數(shù)據(jù)的技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下
2015-04-04
opencv python截取圓形區(qū)域的實(shí)現(xiàn)
本文主要介紹了opencv python截取圓形區(qū)域的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2022-08-08
對(duì)python修改xml文件的節(jié)點(diǎn)值方法詳解
今天小編就為大家分享一篇對(duì)python修改xml文件的節(jié)點(diǎn)值方法詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2018-12-12
python畫圖——實(shí)現(xiàn)在圖上標(biāo)注上具體數(shù)值的方法
今天小編就為大家分享一篇python畫圖——實(shí)現(xiàn)在圖上標(biāo)注上具體數(shù)值的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2019-07-07
Python入門教程(二十六)Python的模塊
這篇文章主要介紹了Python入門教程(二十六)Python的模塊,Python是一門非常強(qiáng)大好用的語(yǔ)言,也有著易上手的特性,本文為入門教程,需要的朋友可以參考下
2023-04-04
在Python中使用第三方模塊的教程
這篇文章主要介紹了在Python中使用第三方模塊的教程,是Python學(xué)習(xí)當(dāng)中的基礎(chǔ)知識(shí),需要的朋友可以參考下
2015-04-04
深入了解Python中frozenset的功能與用法
在Python中,frozenset是一種與set類似的數(shù)據(jù)類型,但具有不可變性,本文將詳細(xì)介紹frozenset的作用、用法以及與set的區(qū)別,感興趣的可以了解下
2024-01-01