在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中函數(shù)的參數(shù)與作用域
這篇文章主要介紹了Python中函數(shù)的參數(shù)與作用域,是Python入門學(xué)習(xí)中的基礎(chǔ)知識,需要的朋友可以參考下2016-03-03
python對Excel按條件進行內(nèi)容補充(推薦)
這篇文章主要介紹了python對Excel按條件進行內(nèi)容補充的相關(guān)知識，非常不錯，本文通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2019-11-11
這篇文章主要介紹了python使用Qt界面以及邏輯實現(xiàn)方法，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07
Tortoise-orm信號實現(xiàn)及使用場景源碼詳解
這篇文章主要為大家介紹了Tortoise-orm信號實現(xiàn)及使用場景源碼詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2023-03-03
設(shè)置jupyter中DataFrame的顯示限制方式
這篇文章主要介紹了設(shè)置jupyter中DataFrame的顯示限制方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來2020-04-04
Python使用APScheduler實現(xiàn)定時任務(wù)過程解析
這篇文章主要介紹了Python使用APScheduler實現(xiàn)定時任務(wù)過程解析,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-09-09
pycharm社區(qū)版安裝node.js插件運行js代碼方法
PyCharm可以說是當(dāng)今最流行的一款Python IDE了,下面這篇文章主要給大家介紹了關(guān)于pycharm社區(qū)版安裝node.js插件運行js代碼的相關(guān)資料,文中通過代碼介紹的非常詳細,需要的朋友可以參考下2023-10-10
這篇文章主要介紹了OpenCV基本操作之傅里葉變換，本文給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-09-09
關(guān)于PyCharm安裝后修改路徑名稱使其可重新打開的問題
這篇文章主要介紹了關(guān)于PyCharm安裝后修改路徑名稱使其可重新打開的問題,本文通過圖文實例相結(jié)合給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-10-10
Windows和Linux下Python輸出彩色文字的方法教程
這篇文章主要介紹了在Windows和Linux中Python輸出彩色文字的方法，通過設(shè)置彩色文字給大家更醒目的效果，文中給出了詳細的介紹和示例代碼，需要的朋友可以參考借鑒，下面來一起看看吧。2017-05-05