zoukankan      html  css  js  c++  java
  • Tensorflow--WIN10系统1070安装GPU版本的Tensorflow1.8.0

    这几天深度学习的过程中发现了一个问题,现在深度学习的过程中用到的tensorflow是CPU版本,而在训练的过程当中所用到的时间感觉特别长,因此想换成gpu版本的tensorflow,来缩短深度学习中模型训练的时间。

    由于之前学习的过程当中所用到的tensorflow是1.0版本,所有写的代码都是基于1.0的版本,因此今天尝试安装的tensorflow是1.0版本的,基于python3.6

    国际惯例,根据这篇博客,开始了安装的过程:

    https://zhuanlan.zhihu.com/p/37086409

    第一步根据步骤,点开我的电脑中的设备管理器,显示适配器中显示显卡为1070

    接着上网百度搜索1070 SPECIFICATION

     

     出来的第一个网页点击进去之后,发现对应的cuda 后面写上对应的数量,因此来说应该是支持的

     接着确定tensorflow型号

    打开cmd控制台

     输入(没有安装过tensorflow的这一步可以省略)

    python

    import tensorflow as tf

    tf.__version__

    结果显示我之前安装的cpu版本的tensorflow是1.2.0的,因此安装的gpu版本应该也要1.0的,不要到2.0,这里先记住,等下再进行搜索安装

    接着点开刚刚英伟达官网的页面下载cuda

    点击图中所示support按钮

     在对应的右边的区域有个技术支持,点击他之后 在对应的上边有对应的developers,选择cuda,然后选择下载就好了,在这里我下载对应的9.0版本--链接

    这里附上详细安装步骤的网址:https://blog.csdn.net/a583621671/article/details/87272803

    依次安装

     win10-1-2-3-4

    安装时间挺长的,耐心等待之后

    安装完成后,在控制台输入nvcc -V

    如果出现以下的画面说明安装成功。

    下载cudnn7.4.1:cudnn下载地址.

    下载需要注册,注册完即可开始下载,之后解压文件,分别放置到

    把其中的bin、include、lib中的文件分别复制到刚才cuda的安装路径:.NVIDIA GPU Computing ToolkitCUDAv9.0下的bin,include,lib中。

    然后在Path添加以下的系统变量:(若不存在就添加)

    C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0in

    C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0include
    C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0libx64
    C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0extrasCUPTIlibx64
    C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0lib

    接着将原本cpu版本的tensorflow覆盖为gpu版本

    pip install --ignore-installed --upgrade tensorflow-gpu

    如若想0安装特定版本的话则执行以下的语句:

    pip install --ignore-installed --upgrade tensorflow_gpu==1.8.0

    耐心等待其安装完成

    测试能否用gpu运行卷积神经网络

     

    运行速度极快

  • 相关阅读:
    WPF 关于拖拽打开文件的注意事项
    asp.net core 3.1中对Mongodb BsonDocument的序列化和反序列化支持
    用百度webuploader分片上传大文件
    多线程学习笔记
    web.config数据库连接字符串加密
    Visual Studio 2010 常用快捷方式
    Team Foundation Server 2013 日常使用使用手册(四)分支与合并
    Team Foundation Server 2013 日常使用使用手册(三)上传新工程、创建任务、创建bug、设置预警
    Team Foundation Server 2013 日常使用使用手册(二)修改、签入、撤销、回滚、对比代码变更
    Team Foundation Server 2013 日常使用使用手册(一)-本地连接TFS、查看任务
  • 原文地址:https://www.cnblogs.com/halone/p/12582458.html
Copyright © 2011-2022 走看看