1、查看tf是gup版本还是cpu版本:
import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
if __name__ == "__main__":
print(device_lib.list_local_devices())
2、查看tf版本号及安装路径:
import tensorflow as tf
if __name__ == "__main__":
print(tf.__version__)
print(tf.__path__)
3、警告:Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
- 如果安装的是GPU版本
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
- 如果安装的是CPU版本
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'