亚洲乱码中文字幕综合,中国熟女仑乱hd,亚洲精品乱拍国产一区二区三区,一本大道卡一卡二卡三乱码全集资源,又粗又黄又硬又爽的免费视频

tensorflow之如何使用GPU而不是CPU問題

 更新時(shí)間：2023年05月10日 16:48:00   作者：叩釘吧zz  
這篇文章主要介紹了tensorflow之如何使用GPU而不是CPU問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

如何使用GPU而不是CPU

首先查看設(shè)備

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())　

如果發(fā)現(xiàn)只有一個(gè)CPU可用

則說明可能存在一下情況： 

1 tensorflow-gpu是否安裝，版本查看，如果版本低于tensorflow，則默認(rèn)啟動(dòng)CPU（t-gpu版本最好高于t）

pip install tensorflow-gpu==2.3.0 -i https://pypi.douban.com/simple/

2 CUDA初始化

在開始訓(xùn)練的前面輸入如下：

# 使用CPU：
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 這一行注釋掉就是使用cpu，不注釋就是使用gpu
# 使用GPU：
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
# os.environ['CUDA_VISIBLE_DEVICES'] = "0" #這個(gè)是僅選擇使用GPU 0
# os.environ['CUDA_VISIBLE_DEVICES'] = "1" #這個(gè)是僅選擇使用GPU 1
# os.environ['CUDA_VISIBLE_DEVICES']  = “0,1” #設(shè)置當(dāng)前使用的GPU設(shè)備為0,1號(hào)

tensorflow使用GPU的設(shè)置方式

方法一

Config=tf.ConfigProto(allow_soft_placement=True) ?##:如果你指定的設(shè)備不存在,允許TF自動(dòng)分配設(shè)備
Config.gpu_options.allow_growth=True ?##動(dòng)態(tài)分配內(nèi)存
sess=tf.session(config=config)

方法二

gpu_options = tf.GPUOptions(allow_growth=True)
gpu_options =tf.GPUOptions(per_process_gpu_memory_fraction=0.8,allow_growth=True) ##每個(gè)gpu占用0.8的顯存
config=tf.ConfigProto(gpu_options=gpu_options,allow_soft_placement=True)
sess=tf.session(config=config)##如果電腦有多個(gè)GPU，tensorflow默認(rèn)全部使用。如果想只使用部分GPU，可以設(shè)置CUDA_VISIBLE_DEVICES。

控制使用哪個(gè)gpu

os.environ[“CUDA_DEVICE_ORDER”] = “PCI_BUS_ID” ##指定的設(shè)備名稱
os.environ[‘CUDA_VISIBLE_DEVICES'] = ‘0' #使用 GPU 0
os.environ[‘CUDA_VISIBLE_DEVICES'] = ‘0,1' # 使用 GPU 0，1

總結(jié)

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

相關(guān)文章

  • Python變量類型知識(shí)點(diǎn)總結(jié)

    Python變量類型知識(shí)點(diǎn)總結(jié)

    在本文里我們給大家分析一篇關(guān)于Python變量類型的相關(guān)知識(shí)點(diǎn)內(nèi)容，需要的朋友們跟著學(xué)習(xí)下。
    2019-02-02
  • Python 自動(dòng)喚醒窗口截圖腳本

    Python 自動(dòng)喚醒窗口截圖腳本

    截圖的操作用途最為廣泛，你可以用它配合定時(shí)工具，定時(shí)檢測(cè)某個(gè)程序的運(yùn)行情況，本文給大家講下如何使用 win32api 實(shí)現(xiàn)自動(dòng)喚醒并截圖的操作，對(duì)Python窗口截圖腳本知識(shí)感興趣的朋友跟隨小編一起看看吧
    2022-02-02
  • Numpy?數(shù)組索引的實(shí)現(xiàn)

    Numpy?數(shù)組索引的實(shí)現(xiàn)

    本文主要介紹了Numpy?數(shù)組索引的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2023-01-01
  • 利用Python繪制一個(gè)可愛的米老鼠

    利用Python繪制一個(gè)可愛的米老鼠

    turtle庫是一個(gè)點(diǎn)線面的簡(jiǎn)單圖像庫，在Python2.6之后被引入進(jìn)來，能夠完成一些比較簡(jiǎn)單的幾何圖像可視化。本文將利用turtle繪制一個(gè)可愛的米老鼠，感興趣的可以試一試
    2022-03-03
  • Python爬取新型冠狀病毒“謠言”新聞進(jìn)行數(shù)據(jù)分析

    Python爬取新型冠狀病毒“謠言”新聞進(jìn)行數(shù)據(jù)分析

    這篇文章主要介紹了Python爬取新型冠狀病毒“謠言”新聞進(jìn)行數(shù)據(jù)分析，本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下
    2020-02-02
  • python使用 cx_Oracle 模塊進(jìn)行查詢操作示例

    python使用 cx_Oracle 模塊進(jìn)行查詢操作示例

    這篇文章主要介紹了python使用 cx_Oracle 模塊進(jìn)行查詢操作,結(jié)合實(shí)例形式分析了Python使用cx_Oracle模塊進(jìn)行數(shù)據(jù)庫的基本連接、查詢、輸出等相關(guān)操作技巧,需要的朋友可以參考下
    2019-11-11
  • Python 爬蟲學(xué)習(xí)筆記之正則表達(dá)式

    Python 爬蟲學(xué)習(xí)筆記之正則表達(dá)式

    正則表達(dá)式是用來匹配字符串非常強(qiáng)大的工具，在其他編程語言中同樣有正則表達(dá)式的概念，Python同樣不例外，利用了正則表達(dá)式，我們想要從返回的頁面內(nèi)容提取出我們想要的內(nèi)容就易如反掌了。
    2016-09-09
  • Python調(diào)用ctypes使用C函數(shù)printf的方法

    Python調(diào)用ctypes使用C函數(shù)printf的方法

    這篇文章主要介紹了Python調(diào)用ctypes使用C函數(shù)printf,需要的朋友可以參考下
    2017-08-08
  • djang常用查詢SQL語句的使用代碼

    djang常用查詢SQL語句的使用代碼

    這篇文章主要介紹了djang常用查詢SQL語句的使用代碼，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2019-02-02
  • Django 實(shí)現(xiàn)外鍵去除自動(dòng)添加的后綴‘_id’

    Django 實(shí)現(xiàn)外鍵去除自動(dòng)添加的后綴‘_id’

    今天小編就為大家分享一篇Django 實(shí)現(xiàn)外鍵去除自動(dòng)添加的后綴‘_id’，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
    2019-11-11

最新評(píng)論