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