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

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

 更新時間：2021年02月20日 11:12:19   作者：-牧野-  
這篇文章主要介紹了詳解tf.device()指定tensorflow運(yùn)行的GPU或CPU設(shè)備實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(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  會打印出執(zhí)行操作所用的設(shè)備，以上輸出：


如果安裝的是GPU版本的tensorflow，機(jī)器上有支持的GPU，也正確安裝了顯卡驅(qū)動、CUDA和cuDNN，默認(rèn)情況下，Session會在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è)備號，統(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)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

相關(guān)文章

  • Python內(nèi)建類型str源碼學(xué)習(xí)

    Python內(nèi)建類型str源碼學(xué)習(xí)

    這篇文章主要為大家介紹了Python內(nèi)建類型str的源碼學(xué)習(xí)，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪
    2022-05-05
  • 詳解如何利用PDoc生成Python文檔

    詳解如何利用PDoc生成Python文檔

    PDoc?是一個強(qiáng)大的?Python?文檔生成工具,它通過解析?Python?代碼中的注釋和類型注解,自動生成格式規(guī)范、內(nèi)容豐富的文檔,下面我們就來看看它的具體使用吧
    2024-11-11
  • django如何根據(jù)現(xiàn)有數(shù)據(jù)庫表生成model詳解

    django如何根據(jù)現(xiàn)有數(shù)據(jù)庫表生成model詳解

    這篇文章主要給大家介紹了關(guān)于django如何根據(jù)現(xiàn)有數(shù)據(jù)庫表生成model的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì),對大家學(xué)習(xí)或者使用Django具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
    2022-08-08
  • Python異步庫asyncio、aiohttp詳解

    Python異步庫asyncio、aiohttp詳解

    這篇文章主要介紹了Python異步庫asyncio、aiohttp使用,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
    2024-06-06
  • python能做哪方面的工作

    python能做哪方面的工作

    在本篇文章里我們給大家整理了關(guān)于學(xué)了python能做哪方面的工作的相關(guān)內(nèi)容，有興趣的朋友們可以學(xué)習(xí)下。
    2020-06-06
  • python如何把字符串類型list轉(zhuǎn)換成list

    python如何把字符串類型list轉(zhuǎn)換成list

    這篇文章主要介紹了python如何吧字符串類型list轉(zhuǎn)換成list,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
    2020-02-02
  • pip命令突然無法使用問題以及解決

    pip命令突然無法使用問題以及解決

    當(dāng)你在使用pip安裝Python庫時遇到問題,可以嘗試以下兩種解決方案,第一種是直接在Scripts文件夾內(nèi)使用CMD命令進(jìn)行安裝,第二種是將Scripts的路徑設(shè)置為系統(tǒng)環(huán)境變量,這樣就可以直接在dos中使用pip install進(jìn)行安裝了,以上解決方案僅供參考,如有更好的方法歡迎交流分享
    2024-10-10
  • 使用numba對Python運(yùn)算加速的方法

    使用numba對Python運(yùn)算加速的方法

    今天小編就為大家分享一篇使用numba對Python運(yùn)算加速的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
    2018-10-10
  • 舉例講解Python裝飾器

    舉例講解Python裝飾器

    這篇文章主要介紹了Python裝飾器的相關(guān)資料，幫助大家更好的理解和使用python，感興趣的朋友可以了解下
    2020-12-12
  • 詳解Python裝飾器由淺入深

    詳解Python裝飾器由淺入深

    裝飾器的功能在很多語言中都有，名字也不盡相同，其實(shí)它體現(xiàn)的是一種設(shè)計模式，強(qiáng)調(diào)的是開放封閉原則，更多的用于后期功能升級而不是編寫新的代碼。本文盡量描述得淺顯易懂，從最基礎(chǔ)的內(nèi)容講起。
    2016-12-12

最新評論