tensorflow指定GPU與動態(tài)分配GPU memory設置

更新時間：2020年02月03日 11:40:18 作者：sherry穎

今天小編就為大家分享一篇tensorflow指定GPU與動態(tài)分配GPU memory設置，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

在tensorflow中，默認指定占用所有的GPU，如需指定占用的GPU，可以在命令行中：

export CUDA_VISIBLE_DEVICES=1

這樣便是只占用1號GPU，通過命令

nvidia-smi

可以查看各個GPU的使用情況。

另外，也可以在python程序中指定GPU，并且動態(tài)分配memory，代碼如下

import os
import sys
os.environ['CUDA_VISIBLE_DEVICES'] = sys.argv[1]
import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
set_session(tf.Session(config=config))

這樣，占用的GPU就是sys.argv[1]了，并且會按需分配memory。

以上這篇tensorflow指定GPU與動態(tài)分配GPU memory設置就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

python 根據(jù)網易云歌曲的ID 直接下載歌曲的實例
今天小編就為大家分享一篇python 根據(jù)網易云歌曲的ID 直接下載歌曲的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-08-08
Python實現(xiàn)線性插值和三次樣條插值的示例代碼
這篇文章主要介紹了Python實現(xiàn)線性插值和三次樣條插值的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2019-11-11
python 巧用正則尋找字符串中的特定字符的位置方法
下面小編就為大家分享一篇python 巧用正則尋找字符串中的特定字符的位置方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-05-05
分享6個隱藏的python功能
給大家詳細分析了6個隱藏的python功能，并詳細講解了每個功能用法，需要的朋友學習下吧。
2017-12-12
詳解python ThreadPoolExecutor異常捕獲
本文主要介紹了詳解python ThreadPoolExecutor異常捕獲，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2023-01-01
python跳出雙層for循環(huán)的解決方法
今天小編就為大家分享一篇python跳出雙層for循環(huán)的解決方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-06-06
python如何讀取指定csv單元格
這篇文章主要介紹了python如何讀取指定csv單元格方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
2024-02-02
Django對數(shù)據(jù)庫進行添加與更新的例子
今天小編就為大家分享一篇Django對數(shù)據(jù)庫進行添加與更新的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-07-07
Django 返回json數(shù)據(jù)的實現(xiàn)示例
這篇文章主要介紹了Django 返回json數(shù)據(jù)的實現(xiàn)示例，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2020-03-03
Python中最強大的錯誤重試庫(tenacity庫)
本文要給大家介紹的tenacity庫，可能是目前Python生態(tài)中最好用的錯誤重試庫，主要介紹tenacity的主要使用方法和特性，具有一定的參考價值，感興趣的可以了解一下
2022-04-04