1. 程式人生 > >linux 運行tensorflow文件缺少_bz2問題及解決

linux 運行tensorflow文件缺少_bz2問題及解決

強行 span init const name 所有 variables 操作 conf

今天,終於把如何在linux服務器上運行tensorflow程序的問題解決:

1、首先要在服務器上python下安裝tensorflow(要看好是在python2還是python3下安裝,還要看好是CPU版本還是GPU版本)

2、確保tensorflow測試程序正常運行,

例:

import tensorflow as tf
 
with tf.device(/cpu:0):
    a = tf.constant([1.0,2.0,3.0],shape=[3],name=a)
    b = tf.constant([1.0,2.0,3.0],shape=[3],name=b)
with tf.device(
/gpu:1): c = a+b #註意:allow_soft_placement=True表明:計算設備可自行選擇,如果沒有這個參數,會報錯。 #因為不是所有的操作都可以被放在GPU上,如果強行將無法放在GPU上的操作指定到GPU上,將會報錯。 sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True)) #sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) sess.run(tf.global_variables_initializer())
print(sess.run(c))

3、確保你寫的程序已經分配GPU或CPU,配置後在服務器上運行即可

tf.device(/cpu:0)
config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True)

 

linux 運行tensorflow文件缺少_bz2問題及解決