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 指定程序在某塊卡上訓練實例就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
今天小編就為大家分享一篇關于python 2 csv寫入空白行的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-06-06
本文主要介紹了python 邊緣擴充方式的實現(xiàn)示例，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-03-03
這篇文章主要介紹了Python中如何將一個類方法變?yōu)槎鄠€方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-12-12
在項目中前后端數(shù)據(jù)相互是一種常態(tài),下面這篇文章主要給大家介紹了關于Django中使用AJAX的詳細過程,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下2022-11-11
這篇文章主要介紹了Python騷操作之動態(tài)定義函數(shù)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-03-03
python作為當前非常受歡迎的編程語言,很大一部分原因是擁有豐富的庫,這篇文章主要給大家介紹了關于Python游戲開發(fā)之精靈和精靈組的相關資料,文中通過圖文介紹的非常詳細,需要的朋友可以參考下2023-05-05
今天小編就為大家分享一篇pytorch 共享參數(shù)的示例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
今天，我想了下，現(xiàn)在把查找PHP WEBSHELL木馬思路發(fā)出來，需要的朋友可以參考下。2011-04-04
本文主要介紹了Django+Vue實現(xiàn)文件上傳下載的項目實踐，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-06-06
Python函數(shù)參數(shù)匹配模型通用規(guī)則keyword-only參數(shù)詳解
Python3對函數(shù)參數(shù)的排序規(guī)則更加通用化了，即Python3 keyword-only參數(shù)，該參數(shù)即為必須只按照關鍵字傳遞而不會有一個位置參數(shù)來填充的參數(shù)。這篇文章主要介紹了Python函數(shù)參數(shù)匹配模型通用規(guī)則keyword-only參數(shù),需要的朋友可以參考下2019-06-06