Windows下安装tensorflow-gpu/cpu教程

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 勾上

四、安装tensorflow

这里使用pip安装,python正常已经自带,但是可能需要更新,根据提示操作即可。

打开cmd,输入pip3 install tensorflow-gpu


安装CPU版:

只需安装python,安装完成后打开cmd, 输入 pip3 install tensorflow 即可


Good Luck!

猜你喜欢

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