當前位置 主頁 > 網站技術 > 代碼類 > 最大化 縮小

    tensorflow:指定gpu 限制使用量百分比,設置最小使用量的實現

    欄目:代碼類 時間:2020-02-06 18:10

    在Python代碼中指定GPU

    import os
    
    os.environ["CUDA_VISIBLE_DEVICES"] = "0"
    

    設置定量的GPU使用量:

    config = tf.ConfigProto()
    
    config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的顯存
    
    session = tf.Session(config=config)
    

    設置最小的GPU使用量:

    config = tf.ConfigProto()
    
    config.gpu_options.allow_growth = True
    
    session = tf.Session(config=config)
    

    以上這篇tensorflow:指定gpu 限制使用量百分比,設置最小使用量的實現就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持IIS7站長之家。

    下一篇:沒有了
777亚洲人成视频免费视频