问题一:查看电脑里tensorflow是gpu还是cpu?
答:在Anaconda Prompt,进入python环境,或者在pycharm中输入:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
结果:
问题二:查看tensorflow版本
答:在Anaconda Prompt,进入python环境,或者在pycharm中输入:
import tensorflow as tf
print(tf.__version__)
结果:
问题三:AttributeError: module ‘tensorflow’ has no attribute 'ConfigProto’类似的问题
原因:代码是基于1.0系列,二系统中的tensorflow是2.0 系列
答:
将下列语句:
config = tf.ConfigProto()
改成
config = tf.compat.v1.ConfigProto()
Session报错也如此:
session = tf.Session(config=config)
改成:
session = tf.compat.v1.Session(config=config)
问题四:出现ModuleNotFoundError: No module named ‘tensorflow.contrib’
答:这个是tensorflow的版本问题,2.0的没有contrib这个模块,所以需要下载2.0的版本,重新装1.x版本,我试验过1.12版本可以。