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

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

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

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

設置使用GPU

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

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 會打印出執(zhí)行操作所用的設備，以上輸出：

如果安裝的是GPU版本的tensorflow，機器上有支持的GPU，也正確安裝了顯卡驅(qū)動、CUDA和cuDNN，默認情況下，Session會在GPU上運行：

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)

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

設置使用cpu

tensorflow中不同的GPU使用/gpu:0和/gpu:1區(qū)分，而CPU不區(qū)分設備號，統(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運行的GPU或CPU設備實現(xiàn)的文章就介紹到這了,更多相關(guān)tensorflow運行GPU或CPU內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

相關(guān)文章

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