windows下安装tensorflow-gpu/cpu

安装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!


猜你喜欢

转载自blog.csdn.net/wang_jiankun/article/details/80137188