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-02
這篇文章主要介紹了Python格式化輸出--%s,%d,%f的代碼詳解，本文通過實例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-04-04
Python獲取暗黑破壞神3戰(zhàn)網(wǎng)前1000命位玩家的英雄技能統(tǒng)計
這篇文章主要介紹了Python獲取暗黑3戰(zhàn)網(wǎng)前1000命位玩家的英雄技能統(tǒng)計的方法,借助urllib2模塊以類似爬蟲的機制來實現(xiàn),需要的朋友可以參考下2016-07-07
這篇文章主要介紹了Python復(fù)制文件操作的方法,涉及Python針對文件與目錄的復(fù)制及刪除操作相關(guān)技巧,具有一定參考借鑒價值,需要的朋友可以參考下2015-11-11
這篇文章主要介紹了Python中IPYTHON的用法入門,較為詳細(xì)的分析了IPYTHON的配置及使用技巧,具有一定參考借鑒價值,需要的朋友可以參考下2015-05-05
Python數(shù)據(jù)分析之matplotlib繪圖詳解
這篇文章主要為大家詳細(xì)介紹了Python數(shù)據(jù)分析之如何利用matplotlib進(jìn)行繪圖，文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以了解一下2022-09-09
這篇文章主要介紹了Django啟動時找不到mysqlclient問題解決方案,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-11-11
python Multiprocessing.Pool進(jìn)程池模塊詳解
multiprocessing模塊提供了一個Process類來代表一個進(jìn)程對象，multiprocessing模塊像線程一樣管理進(jìn)程，這個是multiprocessing的核心，它與threading很相似，對多核CPU的利用率會比threading好的多2022-10-10
python 中Arduino串口傳輸數(shù)據(jù)到電腦并保存至excel表格
這篇文章主要介紹了python Arduino串口傳輸數(shù)據(jù)到電腦并保存至excel表格,本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價值，需要的朋友可以參考下2019-10-10
這篇文章主要介紹了Python Django框架防御CSRF攻擊的方法,結(jié)合實例形式分析了Python Django框架防御CSRF攻擊的原理、配置方法與使用技巧,需要的朋友可以參考下2019-10-10