keras 指定程序在某塊卡上訓練實例
更新時間：2020年06月22日 10:08:57 作者：BYR_jiandong
這篇文章主要介紹了keras 指定程序在某塊卡上訓練實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
場景：某臺機器上有三塊卡，想同時開三個程序，放到三塊卡上去訓練。
策略：CUDA_VISIBLE_DEVICES=1 python train.py就可以指定程序在某塊卡上訓練。
補充知識：keras指定GPU及顯存使用量
指定GPU
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
指定GPU和顯存使用量
import os from keras.backend.tensorflow_backend import set_session os.environ["CUDA_VISIBLE_DEVICES"] = "0" config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.3 set_session(tf.Session(config=config))
指定GPU顯存使用按需分配
import keras.backend.tensorflow_backend as KTF import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" config = tf.ConfigProto() config.gpu_options.allow_growth=True sess = tf.Session(config=config) KTF.set_session(sess)
以上這篇keras 指定程序在某塊卡上訓練實例就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了Python3之字符串比較_重寫cmp函數方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-02-02
下面小編就為大家分享一篇詳談python在windows中的文件路徑問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
Python保存環(huán)境方式(導出requirements.txt文件)
這篇文章主要介紹了Python保存環(huán)境方式(導出requirements.txt文件),具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2025-04-04
這篇文章主要介紹了pandas刪除缺失數據(pd.dropna()方法)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-06-06
基于文件路徑中/?\?//?\\的用法以及絕對相對路徑的問題
這篇文章主要介紹了基于文件路徑中/?\?//?\\的用法以及絕對相對路徑的問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-02-02
今天給大家?guī)淼氖顷P于Python的相關知識,文章圍繞著selenium三種等待方式展開,文中有非常詳細的介紹,需要的朋友可以參考下2021-06-06
python+OpenCV人臉識別考勤系統(tǒng)實現的詳細代碼
作為一個基于人臉識別算法的考勤系統(tǒng)的設計與實現教程，以下內容將提供詳細的步驟和代碼示例。本教程將使用 Python 語言和 OpenCV 庫進行實現,需要的朋友可以參考下2023-05-05
這篇文章主要介紹了Python?selenium八種定位元素的方式，文章圍繞主題展開詳細的內容介紹，具有一定的參考價值，需要的小伙伴可以參考一下2022-08-08
下面小編就為大家分享一篇基于Python的文件類型和字符串詳解，具有很的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2017-12-12
本篇將介紹Python3中的迭代器與生成器，描述可迭代與迭代器關系，并實現自定義類的迭代器模式。非常具有實用價值，需要的朋友可以參考下2018-10-10