Windows上安装 TensorFlow
2021-02-14 21:20
1、官网及帮助文档
官网: https://www.tensorflow.org/install/install_windows
中文帮助文档:https://efeiefei.gitbooks.io/tensorflow_documents_zh/install/install_windows.html
a) cpu版本安装:pip3 install --upgrade tensorflow
b)gpu版本安装:pip3 install --upgrade tensorflow-gpu,需要先安装cuda,cudnn,需要注意版本
备注:目前tensorflow在windows只支持python3.5.x及以上版本,需要使用pip3安装tensorflow。
2、cpu版本安装可能遇到的问题
2.1、ImportError: Could not find ‘msvcp140.dll‘. TensorFlow requires that this DLL... ,解决方案:安装微软为VS2015提供的组件Visual C++ Redistributable for Visual Studio 2015,网址https://www.microsoft.com/zh-CN/download/details.aspx?id=48145,下载后直接安装就好了。
2.2、Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2,应该是编译问题,解决方案:
import os os.environ[‘TF_CPP_MIN_LOG_LEVEL‘] = ‘2‘ #忽略警告
3、简单命令
a)查看tensorflow版本及安装路径
import tensorflow as tf
tf.__version__ #版本
tf.__path__ #路径