溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

keras 指定程序在某塊卡上調(diào)試的方法

發(fā)布時間：2020-06-23 09:28:46 來源：億速云 閱讀：277 作者：清晨 欄目：開發(fā)技術(shù)

不懂keras 指定程序在某塊卡上調(diào)試的方法？其實想解決這個問題也不難，下面讓小編帶著大家一起學(xué)習怎么去解決，希望大家閱讀完這篇文章后大所收獲。

場景：某臺機器上有三塊卡，想同時開三個程序，放到三塊卡上去訓(xùn)練。

策略：CUDA_VISIBLE_DEVICES=1 python train.py就可以指定程序在某塊卡上訓(xùn)練。

keras 指定程序在某塊卡上調(diào)試的方法

補充知識：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 指定程序在某塊卡上調(diào)試的方法內(nèi)容對大家有幫助，同時也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，遇到問題就找億速云，詳細的解決方法等著你來學(xué)習!

向AI問一下細節(jié)

免責聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手