linux 運行tensorflow文件缺少_bz2問題及解決
阿新 • • 發佈:2018-12-21
強行 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問題及解決