zoukankan      html  css  js  c++  java
  • 使用tensorflow下的GPU加速神经网络训练过程

    下载CUDA8.0,安装

    下载cuDNN v5.1安装。放置环境变量等。

    其他版本就不装了。不用找其他版本的关系。

    使用tensorflow-gpu1.0版本。

    使用keras2.0版本。

    有提示的。

    有时候可能需要分配使用空间自动增长:

    config = tf.ConfigProto()
    config.gpu_options.allow_growth = True
    session = tf.Session(config=config)

    英伟达 MX150。华硕FL8000

    TensorFlow版本:1.1.4

    python版本:3.7.3

    CUDA版本:10.0

    和对应的cudnn版本。

    更新地址:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0

  • 相关阅读:
    POJ 1451
    LightOJ 1224
    POJ 2001
    HDU 2072
    POJ 3764
    CH 1602
    CH 1601
    Gym 101873K
    CH 1201
    Gym 101873C
  • 原文地址:https://www.cnblogs.com/bai2018/p/10468628.html
Copyright © 2011-2022 走看看