在tensorflow中設(shè)置使用某一塊GPU、多GPU、CPU的操作
更新時間：2020年02月07日 16:24:11 作者：MachineLP
今天小編就為大家分享一篇在tensorflow中設(shè)置使用某一塊GPU、多GPU、CPU的操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
tensorflow下設(shè)置使用某一塊GPU（從0開始編號）：
import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "1"
多GPU：
num_gpus = 4
for i in range(num_gpus):
with tf.device('/gpu:%d',%i):
。。。
只是用cpu的情況
with tf.device("/cpu:0"):
以上這篇在tensorflow中設(shè)置使用某一塊GPU、多GPU、CPU的操作就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
您可能感興趣的文章:
- 運行tensorflow python程序,限制對GPU和CPU的占用操作
- 基于Tensorflow使用CPU而不用GPU問題的解決
- tensorflow指定CPU與GPU運算的方法實現(xiàn)
- 卸載tensorflow-cpu重裝tensorflow-gpu操作
- 使用Tensorflow-GPU禁用GPU設(shè)置(CPU與GPU速度對比)
- 詳解tf.device()指定tensorflow運行的GPU或CPU設(shè)備實現(xiàn)
- 用gpu訓(xùn)練好的神經(jīng)網(wǎng)絡(luò),用tensorflow-cpu跑出錯的原因及解決方案
- Tensorflow中使用cpu和gpu有什么區(qū)別
- tensorflow之如何使用GPU而不是CPU問題
- TensorFlow安裝CPU版本和GPU版本的實現(xiàn)步驟
相關(guān)文章
這篇文章主要介紹了Python線程,這篇開始我們將進入中級編程。處理更加復(fù)雜事情。比如本文的線程，咱們先從基礎(chǔ)知識入手,需要的朋友可以參考下下面文章的詳細內(nèi)容2022-02-02
這篇文章主要介紹了python 爬取小說并下載的示例，幫助大家更好的理解和學習python爬蟲，感興趣的朋友可以了解下2020-12-12
這篇文章主要為大家詳細介紹了PyQt5每天必學之布局管理的相關(guān)資料，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-04-04
這篇文章主要為大家詳細介紹了python實現(xiàn)二維數(shù)組的對角線遍歷，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-03-03
Python函數(shù)的定義方式與函數(shù)參數(shù)問題實例分析
這篇文章主要介紹了Python函數(shù)的定義方式與函數(shù)參數(shù)問題,結(jié)合實例形式詳細分析了Python函數(shù)定義、函數(shù)參數(shù)相關(guān)原理、操作技巧與注意事項,需要的朋友可以參考下2019-12-12
Rich庫的功能就像它的名字一樣,使Python編程更加豐富（rich）,用來幫助開發(fā)者在控制臺（命令行）輸出中創(chuàng)建豐富、多彩和具有格式化的文本,下面我們就來了解下它的具體使用吧2024-02-02
感覺好久沒和大家一起寫小游戲玩了,今天恰巧有空.這次我們來用Python做個2048小游戲吧.廢話不多說,文中有非常詳細的代碼示例,需要的朋友可以參考下2021-06-06
這篇文章主要為大家詳細介紹了Python實現(xiàn)雙向鏈表，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-05-05
簡要講解Python編程中線程的創(chuàng)建與鎖的使用
這篇文章主要介紹了簡要講解Python編程中線程的創(chuàng)建與鎖的使用,Python中雖然有GIL的存在,但依然是能夠創(chuàng)建多個線程來交替使用的,需要的朋友可以參考下2016-02-02
這篇文章主要介紹了python requests庫的使用，幫助大家更好的利用python進行爬蟲，感興趣的朋友可以了解下2021-01-01