安装GPU版:
一、安装cuda
在官网上下载对应的版本:https://developer.nvidia.com/cuda-downloads
安装过程一路next,最好不要修改安装路径。正常会自己添加PATH
PS:tensorflow目前不支持cuda9.1及以上
二、安装cuDNN
在官网上下载:https://developer.nvidia.com/rdp/cudnn-download
下载解压缩后,将文件夹中的cndnn64_7.dll、cudnn.h、cudnn.lib分别复制到
C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin
C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.1\include
C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.1\lib\x64
注意:cudnn需要与cuda版本相对应。
三、安装python
在官网上下载对应的安装包:https://www.python.org
双击安装即可:记得把 Add Python xx to PATH 勾上
python正常已经自带pip,可能需要更新
四、安装tensorflow
打开cmd,输入pip3 install tensorflow-gpu
-----------------------------------------------------------------------------------------------------------
安装CPU版:
只需安装python,安装完成后打开cmd, 输入 pip3 install tensorflow 即可
-----------------------------------------------------------------------------------------------------------
good luck!