tensorflow:指定gpu 限制使用量百分比,設(shè)置最小使用量的實現(xiàn)
更新時間:2020年02月06日 15:23:10 作者:ming.zhang
今天小編就為大家分享一篇tensorflow:指定gpu 限制使用量百分比,設(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è)置最小使用量的實現(xiàn)就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
淺談spring boot 集成 log4j 解決與logback沖突的問題
今天小編就為大家分享一篇淺談spring boot 集成 log4j 解決與logback沖突的問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02Python獲取暗黑破壞神3戰(zhàn)網(wǎng)前1000命位玩家的英雄技能統(tǒng)計
這篇文章主要介紹了Python獲取暗黑3戰(zhàn)網(wǎng)前1000命位玩家的英雄技能統(tǒng)計的方法,借助urllib2模塊以類似爬蟲的機制來實現(xiàn),需要的朋友可以參考下2016-07-07Python數(shù)據(jù)分析之matplotlib繪圖詳解
這篇文章主要為大家詳細介紹了Python數(shù)據(jù)分析之如何利用matplotlib進行繪圖,文中的示例代碼講解詳細,感興趣的小伙伴可以了解一下2022-09-09python Multiprocessing.Pool進程池模塊詳解
multiprocessing模塊提供了一個Process類來代表一個進程對象,multiprocessing模塊像線程一樣管理進程,這個是multiprocessing的核心,它與threading很相似,對多核CPU的利用率會比threading好的多2022-10-10python 中Arduino串口傳輸數(shù)據(jù)到電腦并保存至excel表格
這篇文章主要介紹了python Arduino串口傳輸數(shù)據(jù)到電腦并保存至excel表格,本文給大家介紹的非常詳細,具有一定的參考借鑒價值,需要的朋友可以參考下2019-10-10