tensorflow安装相关问题

安装步骤

要求:Python必须是64位
根据TensorFlow的计算方式,TensorFlow的安装分为CPU版本和GPU版本对于Python3.5或者Python3.6,可以使用pip install tensorflow(安装CPU版本)和pip install tensorflow-gpu(安装GPU版本)对于Python2.7,只能通过源码编译来安装TensorFlow(Windows操作系统)

备注:TensorFlow-GPU要求机器的显卡必须是NVidia的显卡。
备注:授课TensorFlow版本选择1.4.0

安装命令:
pip install tensorflow==1.4.0

TensorFlow GPU版本安装:
环境:Python 3.6、NVIDIA显卡
安装过程:
安装CUDA SDK(默认会将bin文件夹添加到PATH环境变量中,即安装CUDA和cuDNN)

安装TensorFlow GPU,

安装命令:

pip install tensorflow-gpu==1.4.0

CUDA下载安装链接:https://developer.nvidia.com/cuda-toolkit-archive
cuDNN下载安装链接:https://developer.nvidia.com/rdp/cudnn-archive

备注:具体的CUDA版本根据导入tensorflow时提示的异常来选择;即先安装tensorflow-gpu,然后在python的命令行执行:import tensorflow,会出现如下异常,则表示我们需要安装的是CUDA 8.0版本,至于cuDNN选择和CUDA对应版本即可。

除了安装CUDA外,还需要安装对应版本的cuDNN,将cuDNN的压缩包解压,然后将压缩包中的三个文件夹全部放到CUDA对应的根目录下,根据环境变量找到对应的根目录:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0;

 

tensorflow安装后,按照如下代码运行,发现报错:

该原因一般是需要下载源码,

https://codeload.github.com/tensorflow/tensorflow/zip/master

下载后放到python的lib目录下D:PythonPython36Lib

原文地址:https://www.cnblogs.com/yuluoxingkong/p/9053194.html