gpu='0' #表示pci卡槽gpu0可見
gpu='3' #表示pci卡槽gpu3可見
gpu='3,2,0' #表示pci卡槽gpu3,2,0可見，gpu0不可見
os.environ['cuda_visible_devices'] = gpu #pci卡槽多gpu可見設定
gpu = '-1' #表示gpu不可見，即only cpu模式
os.environ['cuda_visible_devices'] = gpu #pci卡槽gpu不可見設定
num_gpus = 4 #num_gpus =表示同時使用gpu個數
num_cpus = 1 #num_cpus =表示同時使用cpu個數,且num_cpus>=1
config = tf.configproto(
device_count=,
allow_soft_placement=true, #自動選中gpu
log_device_placement=false #列印裝置分配日誌
)self.sess = tf.session(config=config)
num_gpus = 0
num_cpus = 1 #num_cpus =表示同時使用cpu個數,且num_cpus>=1
config = tf.configproto(
device_count=,
allow_soft_placement=true, #自動選中gpu
log_device_placement=false #列印裝置分配日誌
)self.sess = tf.session(config=config)
tensorflow中模型的儲存
sess.run fetches,feed dict none,options none,run metadata none tf.session.run 執行 fetches 中的操作，計算 fetches 中的張量值。這個函式執行一步 tensorflow 運算，通過執行必要的圖塊來執行每乙個操...
TensorFlow中的GNMT模型構建大致過程
tensorflow中gnmt的實現採取多層雙向lstm構建，構建基本過程如下 encoder bi output,bi state tf.nn.bidirectional dynamic rnn tf.nn.bidirectional dynamic rnn fw cell,bw cell,inp...
tensorflow 檢視ckpt模型中引數值
有時我們有檢視tensor內部變數的值的變化情況，需要挖一下，現給出解析 date 2021 5 18 author import osfrom tensorflow.python import pywrap tensorflow base model dir r model ckp 2021022...