欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

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

 更新時間：2020年02月03日 11:40:18   作者：sherry穎  
今天小編就為大家分享一篇tensorflow指定GPU與動態(tài)分配GPU memory設(shè)置，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧

在tensorflow中，默認(rèn)指定占用所有的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設(shè)置就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

相關(guān)文章

  • python 根據(jù)網(wǎng)易云歌曲的ID 直接下載歌曲的實(shí)例

    python 根據(jù)網(wǎng)易云歌曲的ID 直接下載歌曲的實(shí)例

    今天小編就為大家分享一篇python 根據(jù)網(wǎng)易云歌曲的ID 直接下載歌曲的實(shí)例，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
    2019-08-08
  • Python實(shí)現(xiàn)線性插值和三次樣條插值的示例代碼

    Python實(shí)現(xiàn)線性插值和三次樣條插值的示例代碼

    這篇文章主要介紹了Python實(shí)現(xiàn)線性插值和三次樣條插值的示例代碼，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2019-11-11
  • python 巧用正則尋找字符串中的特定字符的位置方法

    python 巧用正則尋找字符串中的特定字符的位置方法

    下面小編就為大家分享一篇python 巧用正則尋找字符串中的特定字符的位置方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
    2018-05-05
  • 分享6個隱藏的python功能

    分享6個隱藏的python功能

    給大家詳細(xì)分析了6個隱藏的python功能，并詳細(xì)講解了每個功能用法，需要的朋友學(xué)習(xí)下吧。
    2017-12-12
  • 詳解python ThreadPoolExecutor異常捕獲

    詳解python ThreadPoolExecutor異常捕獲

    本文主要介紹了詳解python ThreadPoolExecutor異常捕獲，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2023-01-01
  • python跳出雙層for循環(huán)的解決方法

    python跳出雙層for循環(huán)的解決方法

    今天小編就為大家分享一篇python跳出雙層for循環(huán)的解決方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
    2019-06-06
  • python如何讀取指定csv單元格

    python如何讀取指定csv單元格

    這篇文章主要介紹了python如何讀取指定csv單元格方式,具有很好的參考價(jià)值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
    2024-02-02
  • Django對數(shù)據(jù)庫進(jìn)行添加與更新的例子

    Django對數(shù)據(jù)庫進(jìn)行添加與更新的例子

    今天小編就為大家分享一篇Django對數(shù)據(jù)庫進(jìn)行添加與更新的例子，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
    2019-07-07
  • Django 返回json數(shù)據(jù)的實(shí)現(xiàn)示例

    Django 返回json數(shù)據(jù)的實(shí)現(xiàn)示例

    這篇文章主要介紹了Django 返回json數(shù)據(jù)的實(shí)現(xiàn)示例，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2020-03-03
  • Python中最強(qiáng)大的錯誤重試庫(tenacity庫)

    Python中最強(qiáng)大的錯誤重試庫(tenacity庫)

    本文要給大家介紹的tenacity庫，可能是目前Python生態(tài)中最好用的錯誤重試庫，主要介紹tenacity的主要使用方法和特性，具有一定的參考價(jià)值，感興趣的可以了解一下
    2022-04-04

最新評論