• windows10配置tensorflow环境


    1、配置流程概览

    1.配置cuda与cudnn
    2.配置anaconda
    3.在anaconda中配置tensorflow

    1.1 什么是CUDA和Cudnn

    首先我们要明白几个概念以及他们之间的关系:显卡驱动、CUDA以及Cudnn

    • 显卡驱动:显卡驱动就是用来驱动显卡的程序,它是硬件所对应的软件。打一个不是很恰当的比喻,操作系统就像一个人,显卡就比作一辆车,显卡驱动就像车钥匙,没有对应的驱动,操作系统就无法调用对应的硬件。
    • CUDA:CUDA是NVIDIA推出的用于自家GPU的并行计算框架,也就是说CUDA只能在NVIDIA的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。CUDA的本质是一个工具包(ToolKit)
    • Cudnn:NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中。
    • CUDA与CUDNN的关系
          CUDA看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手。但是CUDA这个工作台买来的时候,并没有送扳手。想要在CUDA上运行深度神经网络,就要安装cuDNN,就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作,工作速度相较CPU快很多。
      cuDNN是一个SDK,是一个专门用于神经网络的加速包,注意,它跟我们的CUDA没有一一对应的关系,即每一个版本的CUDA可能有好几个版本的cuDNN与之对应,但一般有一个最新版本的cuDNN版本与CUDA对应更好。
    • CUDA与显卡驱动
      NVIDIA的显卡驱动器与CUDA并不是一一对应,CUDA本质上只是一个工具包而已,所以我可以在同一个设备上安装很多个不同版本的CUDA工具包。一般情况下,我只需要安装最新版本的显卡驱动,然后根据自己的选择选择不同CUDA工具包就可以。但是使用的CUDA版本不能高于自己的显卡驱动版本。

    1.2 tensorflow各个版本的CUDA以及Cudnn版本对应关系

    相应的网址为:
    https://www.tensorflow.org/install/source#common_installation_problems
    https://www.tensorflow.org/install/source_windows

    版本 Python 版本 编译器 编译工具 cuDNN CUDA
    tensorflow_gpu-2.0.0-alpha0 2.7、3.3-3.6 GCC 4.8 Bazel 0.19.2 7.4.1以及更高版本 CUDA 10.0 (需要 410.x 或更高版本)
    tensorflow_gpu-1.13.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.19.2 7.4 10.0
    tensorflow_gpu-1.12.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.15.0 7 9
    tensorflow_gpu-1.11.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.15.0 7 9
    tensorflow_gpu-1.10.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.15.0 7 9
    tensorflow_gpu-1.9.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.11.0 7 9
    tensorflow_gpu-1.8.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.10.0 7 9
    tensorflow_gpu-1.7.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.9.0 7 9
    tensorflow_gpu-1.6.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.9.0 7 9
    tensorflow_gpu-1.5.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.8.0 7 9
    tensorflow_gpu-1.4.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.5.4 6 8
    tensorflow_gpu-1.3.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.4.5 6 8
    tensorflow_gpu-1.2.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.4.5 5.1 8
    tensorflow_gpu-1.1.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.4.2 5.1 8
    tensorflow_gpu-1.0.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.4.2 5.1 8

    2、具体配置

    2.1 CUDA与CUDNN

    CUDA的下载地址为:https://developer.nvidia.com/cuda-downloads

    2.2 CUDNN

    CUDnn的下载地址为:https://developer.nvidia.com/rdp/cudnn-archive


    2.3 Anaconda

    Anaconda的下载地址为:https://developer.nvidia.com/cuda-downloads

    这里将两个都勾选,然后一路next

    2.4 tensorflow

    1. 打开Anaconda Prompt,创建虚拟环境:conda create -n tensorflow python=3.6
    2. 激活tensorflow环境 activate tensorflow
    3. 安装tensorflow
      这里介绍两种方法
    • 如果使用比较新的tensorflow-gpu版本,使用
    pip install tensorflow-gpu==1.14 -i https://pypi.tuna.tsinghua.edu.cn/simple
    //如果清华源不行 豆瓣源比较好用
    
    • 如果是比较老版本的(例如tensorflow-gpu==1.4)
    1. 搜索tf
      anaconda search -t conda tensorflow-gpu
      接下来选择一个源地址

    2. 安装tf
      conda install --channel https://conda.anaconda.org/aaronzs tensorflow-gpu=1.4.0

      最后一行会出现安装方式,如图所示

    2.5 测试是否配置成功

    输入python,进入python命令行

    3、遇到的问题以及解决方案

    conda常用的命令。
    1)conda list 查看安装了哪些包。
    2)conda env list 或 conda info -e 查看当前存在哪些虚拟环境

    • 虚拟环境装烂了
      使用命令conda remove -n your_env_name(虚拟环境名称) --all, 即可删除环境(釜底抽薪)。
      跑tensorflow时遇到OOM问题

      选择降低每一块的大小

    4、参考文章

    tensorflow各个版本的CUDA以及Cudnn版本对应关系
    CUDA与cuDNN
    https://blog.csdn.net/u012620515/article/details/78792670

  • 相关阅读:
    shell脚本根据端口号kill掉进程
    使用netstat -ano 查看机器端口的占用情况(windows环境)
    分享一两个小工具,
    将压缩文件伪装图片格式文件以及将python文件转化为exe文件(测试完,真的有效)
    celery 异步任务 周期任务 定时任务的实现
    wsgi、uwsgi、asgi协议的关系
    centos7忘记密码更改步骤
    工作遇到的坑以及自己的学习悟道之道
    案例小集锦
    asp.net mvc部署
  • 原文地址:https://www.cnblogs.com/wjune-0405/p/14782598.html
Copyright © 2020-2023  润新知