linux 執行tensorflow檔案缺少_bz2問題及解決
阿新 • • 發佈:2018-12-20
今天,終於把如何在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)