美文网首页
记-深度学习-Win7安装GPU版的TensorFlow

记-深度学习-Win7安装GPU版的TensorFlow

作者: Andrew0000 | 来源:发表于2019-10-23 17:23 被阅读0次
    查看电脑显卡是否被官网支持

    可以参考 这个

    明确CUDA、CuDNN以及tf-gpu版本对应

    官网

    笔者选择的套餐为:tf-1.12.0、cuDNN7、CUDA9

    注意:安装tf-gpu默认你已经安装好Anaconda、以及 Microsoft Visual C++ 2015 Redistributable 更新 3等相关 C语言库

    官网下载对应CUDA版本

    在这个网址找到你希望安装的版本下载即可,注意操作系统、版本、下载文本类型(离线还是联网)

    官网下载对应于CUDA的CuDNN

    在这个网址下载对应CUDA9的CuDNN7,注意这里需要注册,而且网速慢,原因你懂的。

    安装CUDA

    直接打开exe文件,一路下一步即可。当然可以选择自定义,不过具体没研究!

    解压CuDNN

    待CUDA安装完成,在默认目录下 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0可以看到bin、lib目录

    同样,对于CuDNN解压发现目录结构同上类似,这里把解压后文件夹里面对应的文件(注意是文件复制过去,不是目录)依次复制给CUDA上述目录,注意没有覆盖操作!

    添加环境变量

    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin;

    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64;

    将上述两个目录(此时该目录下应该有CuDNN的文件,其实我理解就是CUDA的补丁)添加进入环境变量

    检查安装情况

    nvcc -V

    显示你安装的CUDA工具包,就是安装成功。

    创建tf-gpu环境并且安装对应版本tf

    conda create -n tf-gpu python=3.6

    activate tf-gpu

    pip install tensorflow-gpu==1.12.0

    至此,tf-gpu版本安装结束

    测试是否安装成功

    **import tensorflow as tf hello=tf.constant(‘hello,world’) sess=tf.Session() print(sess.run(hello))**

    安装jupyter notebook

    参考这个

    如果安装后打开,需要用到密码!

    jupyter notebook password

    键入之后输入两次密码,输入过程不可见!

    之后打开jupyter notebook即可使用

    总结

    主要注意上述各版本对应情况,千万不要偷懒,认真对比选择你的套餐!

    其次就是C语言环境,VC++2015需要有!

    参考及进阶

    入坑安装参考

    进阶学习指南

    相关文章

      网友评论

          本文标题:记-深度学习-Win7安装GPU版的TensorFlow

          本文链接:https://www.haomeiwen.com/subject/afwkvctx.html