Windows10下搭建TensorFlow环境(GPU版本)

软件清单:

安装环境:Windows10_64位

Python:3.6

TensorFlow:1.8.0

CUDA:CUDA9.0

cuDNN:cuDNN7.0

安装过程

1.安装python

建议安装Anaconda科学计算平台来管理你的python环境,里面自带了很多模板,下载很简单,直接下一步就好,这里就不赘述。

(若之前有其它的python版本,最好卸载了,以免不必要的麻烦)

官方下载:https://www.anaconda.com/download/

2.安装TensorFlow

安装好上述的python3.6环境后,在cmd中

输入:pip install tensorflow-gpu==1.8.0

这里默认的是下载最新版本的,本次安装是2018年7月8日下载,版本号是1.8.0。(注意,tensorflow有多个版本,可能会有多种安装方式,一定要找到对应的cuda和cuDNN版本下载,可以参照这里的安装版本)

3.安装cuda

安装cuda的时候一定要安装与当前tensorflow版本相匹配的cuda版本,这里安装的cuda版本应该是9.0的。

官方地址:https://developer.nvidia.com/cuda-toolkit-archive





(双击就可以进行安装,注意安装的时候记得关掉杀毒软件,比如360!之前一直安装失败,花了很多时间就是因为被它拦截了,一直下一步下载就好)

4.安装cuDNN7

安装cuDNN库文件,这里可能需要填一个小问卷,随便填就好,下载的是一个压缩包。(下载之前需要注册账号,注册过程相当简单)

下载地址:https://developer.nvidia.com/rdp/cudnn-download





(找到对应的版本,最好使用7.0版本的,比较稳定吧)

5.关联CUDA与cuDNN

解压cuDNN压缩包后,把解压后的文件分别复制粘贴到cuda对应的文件夹里。




(到此就完成了所有的配置了)

6.测试

直接用cmd进入python命令行界面

输入:import tensorflow

sess = tensorflow.Session()

就可以看到如下信息。



(可以看到显卡信息940M。。。。。。很垃圾的显卡,将就用吧)

到这里就大功告成了。


猜你喜欢

转载自blog.csdn.net/qq_41004007/article/details/80961484