tensorflow:指定gpu 限制使用量百分比,設(shè)置最小使用量的實(shí)現(xiàn)
在Python代碼中指定GPU
import os os.environ["CUDA_VISIBLE_DEVICES"] = "0"
設(shè)置定量的GPU使用量:
config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的顯存 session = tf.Session(config=config)
設(shè)置最小的GPU使用量:
config = tf.ConfigProto() config.gpu_options.allow_growth = True session = tf.Session(config=config)
以上這篇tensorflow:指定gpu 限制使用量百分比,設(shè)置最小使用量的實(shí)現(xiàn)就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
- 基于Tensorflow使用CPU而不用GPU問題的解決
- 運(yùn)行tensorflow python程序,限制對GPU和CPU的占用操作
- tensorflow使用指定gpu的方法
- 基于tensorflow指定GPU運(yùn)行及GPU資源分配的幾種方式小結(jié)
- tensorflow指定GPU與動態(tài)分配GPU memory設(shè)置
- 檢測tensorflow是否使用gpu進(jìn)行計(jì)算的方式
- 解決TensorFlow GPU版出現(xiàn)OOM錯誤的問題
- 解決Tensorflow占用GPU顯存問題
- 已安裝tensorflow-gpu,但keras無法使用GPU加速的解決
相關(guān)文章
淺談spring boot 集成 log4j 解決與logback沖突的問題
今天小編就為大家分享一篇淺談spring boot 集成 log4j 解決與logback沖突的問題,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02Python獲取暗黑破壞神3戰(zhàn)網(wǎng)前1000命位玩家的英雄技能統(tǒng)計(jì)
這篇文章主要介紹了Python獲取暗黑3戰(zhàn)網(wǎng)前1000命位玩家的英雄技能統(tǒng)計(jì)的方法,借助urllib2模塊以類似爬蟲的機(jī)制來實(shí)現(xiàn),需要的朋友可以參考下2016-07-07Python數(shù)據(jù)分析之matplotlib繪圖詳解
這篇文章主要為大家詳細(xì)介紹了Python數(shù)據(jù)分析之如何利用matplotlib進(jìn)行繪圖,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以了解一下2022-09-09python Multiprocessing.Pool進(jìn)程池模塊詳解
multiprocessing模塊提供了一個Process類來代表一個進(jìn)程對象,multiprocessing模塊像線程一樣管理進(jìn)程,這個是multiprocessing的核心,它與threading很相似,對多核CPU的利用率會比threading好的多2022-10-10python 中Arduino串口傳輸數(shù)據(jù)到電腦并保存至excel表格
這篇文章主要介紹了python Arduino串口傳輸數(shù)據(jù)到電腦并保存至excel表格,本文給大家介紹的非常詳細(xì),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-10-10