因為涉及到同一臺電腦多個GPU，在指定tensorflow圖的時候，需要為不同的圖指定不同的GPU，所以必須在導入tensorflow之前，指定可用的GPU

def import_tf(device_id=-1, verbose=False):"""import tensorflow, set tensorflow graph load device, set tensorflow log level, return tensorflow instance:param device_id: GPU id:param verbose: tensorflow logging level:return: tensorflow instance"""# set visible gpu, -1 is cpuos.environ['CUDA_VISIBLE_DEVICES'] = '-1' if device_id < 0 else str(device_id)os.environ['TF_CPP_MIN_LOG_LEVEL'] = '0' if verbose else '3'import tensorflow as tftf.logging.set_verbosity(tf.logging.DEBUG if verbose else tf.logging.ERROR)return tf

?

轉載于:https://www.cnblogs.com/callyblog/p/10388454.html

《新程序員》：云原生和全面數字化實踐50位技術專家共同創作，文字、視頻、音頻交互閱讀

總結

以上是生活随笔為你收集整理的tensorflow 导入新的tensorflow实例的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。