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

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

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

如何使用GPU而不是CPU

首先查看設(shè)備

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

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

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

1 tensorflow-gpu是否安裝，版本查看，如果版本低于tensorflow，則默認(rèn)啟動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" #這個是僅選擇使用GPU 0
# os.environ['CUDA_VISIBLE_DEVICES'] = "1" #這個是僅選擇使用GPU 1
# os.environ['CUDA_VISIBLE_DEVICES']  = “0,1” #設(shè)置當(dāng)前使用的GPU設(shè)備為0,1號

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

方法一

Config=tf.ConfigProto(allow_soft_placement=True) ?##:如果你指定的設(shè)備不存在,允許TF自動分配設(shè)備
Config.gpu_options.allow_growth=True ?##動態(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) ##每個gpu占用0.8的顯存
config=tf.ConfigProto(gpu_options=gpu_options,allow_soft_placement=True)
sess=tf.session(config=config)##如果電腦有多個GPU，tensorflow默認(rèn)全部使用。如果想只使用部分GPU，可以設(shè)置CUDA_VISIBLE_DEVICES。

控制使用哪個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é)

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

相關(guān)文章

  • Python變量類型知識點總結(jié)

    Python變量類型知識點總結(jié)

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

    Python 自動喚醒窗口截圖腳本

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

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

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

    利用Python繪制一個可愛的米老鼠

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

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

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

    python使用 cx_Oracle 模塊進行查詢操作示例

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

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

    正則表達式是用來匹配字符串非常強大的工具，在其他編程語言中同樣有正則表達式的概念，Python同樣不例外，利用了正則表達式，我們想要從返回的頁面內(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語句的使用代碼，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2019-02-02
  • Django 實現(xiàn)外鍵去除自動添加的后綴‘_id’

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

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

最新評論