1. 程式人生 > >linux 執行tensorflow檔案缺少_bz2問題及解決

linux 執行tensorflow檔案缺少_bz2問題及解決

今天,終於把如何在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)