zoukankan      html  css  js  c++  java
  • win10 + cuda(v9.0) 安装TensorFlow-gpu版

    之前在实习公司的电脑上装过TensorFlow-gpu,那时候很快就装好了。但在自己的笔记本上装时,却搞了很久。。。
    一部分原因是因为用校园网下载cuda toolkit 和cudnn ,总是在最后时刻罢工。辗转叫了几个人帮忙,才把这两个东西下载好。。

    安装步骤

    其实,这个安装过程还是蛮简单的。据说cuda版本和cudnn版本没选好的话,就会遇到很多坑,所幸,我选的版本都能很好的驾驭tensorflow。

    下面是要安装的东西,可以依次安装,也可以按你喜欢的顺序来安装。但要保证安装了VC++插件和python之后,才安装tensorflow。

    • Visual C++ Redistributable for Visual Studio 2015: tensorflow其实是基于VC++2015开发的,所以需要安装Visual C++ Redistributable for Visual Studio 2015。下载后,直接默认安装就行

    • python环境:我用的是anaconda3.x,新建了一个python3.5的环境,然后在python3.5中安装的tensorflow-gpu。这个不会的话,可以尝试搜索“anaconda安装、anaconda清华源安装”和“anaconda新建python环境”。。

    • cuda9.0: cuda-v9.0.下载之后,默认安装就行,中间要选同意的地方就选同意。

    这里写图片描述

    • 安装好cuda后,要配置系统环境。用小娜搜索“控制面板”–>选择大图标–>选择系统–>选择高级系统设置–>选择环境变量–>选择path,新增加“C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0”–>确定。。然后再新增几个变量,如下图(已经存在的就不用增加了):
      这里写图片描述

    • cudnn:cuDNN v7.1.4 Library for Windows 10) 下载好后,解压,把所有东西复制到C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0 中,就ok啦

    • tensorflow-gpu:安装win+R,运行cmd,打开命令窗口,激活python3.5的环境(activate 环境名),然后输入pip install tensorflow-gpu 即可。等待安装成功就行。

    测试

    我喜欢用jupyter notebook来写python,所以现在也用它来写一下测试代码:

    import tensorflow as tf
    
    hi = tf.constant('Hi , welcome to TensorFlow ! ')
    sess = tf.Session()
    print(sess.run(hi).decode())

    得到结果如下:
    这里写图片描述

    cool,顺利输出了结果,说明安装成功。
    这里写图片描述

  • 相关阅读:
    Linux下的lds链接脚本简介
    Fedora下载地址
    SkyEye的使用
    shell变量详解
    Linux shell 脚本攻略之正则表达式入门
    Linux shell 脚本攻略之统计文件的行数、单词数和字符数
    Linux shell 脚本攻略之创建不可修改文件
    Linux shell 脚本攻略之生成任意大小的文件
    Linux shell 脚本攻略之批量重命名
    Linux shell 脚本攻略之文件查找与文件列表
  • 原文地址:https://www.cnblogs.com/surecheun/p/9648949.html
Copyright © 2011-2022 走看看