快捷導(dǎo)航

在tensorflow中設(shè)置使用某一塊GPU、多GPU、CPU的操作

更新時(shí)間：2020年02月07日 16:24:11 作者：MachineLP

今天小編就為大家分享一篇在tensorflow中設(shè)置使用某一塊GPU、多GPU、CPU的操作，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧

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)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python基礎(chǔ)學(xué)習(xí)之認(rèn)識線程
這篇文章主要介紹了Python線程,這篇開始我們將進(jìn)入中級編程。處理更加復(fù)雜事情。比如本文的線程，咱們先從基礎(chǔ)知識入手,需要的朋友可以參考下下面文章的詳細(xì)內(nèi)容
2022-02-02
python 爬取小說并下載的示例
這篇文章主要介紹了python 爬取小說并下載的示例，幫助大家更好的理解和學(xué)習(xí)python爬蟲，感興趣的朋友可以了解下
2020-12-12
PyQt5每天必學(xué)之布局管理
這篇文章主要為大家詳細(xì)介紹了PyQt5每天必學(xué)之布局管理的相關(guān)資料，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2018-04-04
python實(shí)現(xiàn)二維數(shù)組的對角線遍歷
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)二維數(shù)組的對角線遍歷，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2019-03-03
Python函數(shù)的定義方式與函數(shù)參數(shù)問題實(shí)例分析
這篇文章主要介紹了Python函數(shù)的定義方式與函數(shù)參數(shù)問題,結(jié)合實(shí)例形式詳細(xì)分析了Python函數(shù)定義、函數(shù)參數(shù)相關(guān)原理、操作技巧與注意事項(xiàng),需要的朋友可以參考下
2019-12-12
Python使用Rich實(shí)現(xiàn)美化終端顯示效果
Rich庫的功能就像它的名字一樣,使Python編程更加豐富（rich）,用來幫助開發(fā)者在控制臺（命令行）輸出中創(chuàng)建豐富、多彩和具有格式化的文本,下面我們就來了解下它的具體使用吧
2024-02-02
用Python手把手教你實(shí)現(xiàn)2048小游戲
感覺好久沒和大家一起寫小游戲玩了,今天恰巧有空.這次我們來用Python做個(gè)2048小游戲吧.廢話不多說,文中有非常詳細(xì)的代碼示例,需要的朋友可以參考下
2021-06-06
Python實(shí)現(xiàn)雙向鏈表
這篇文章主要為大家詳細(xì)介紹了Python實(shí)現(xiàn)雙向鏈表，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2022-05-05
簡要講解Python編程中線程的創(chuàng)建與鎖的使用
這篇文章主要介紹了簡要講解Python編程中線程的創(chuàng)建與鎖的使用,Python中雖然有GIL的存在,但依然是能夠創(chuàng)建多個(gè)線程來交替使用的,需要的朋友可以參考下
2016-02-02
python requests庫的使用
這篇文章主要介紹了python requests庫的使用，幫助大家更好的利用python進(jìn)行爬蟲，感興趣的朋友可以了解下
2021-01-01