美文网首页
Ubuntu下theanorc文件配置

Ubuntu下theanorc文件配置

作者: MarkSJL | 来源:发表于2017-03-05 09:38 被阅读0次

    配置这个文件的目的是,可以切换cpu和gpu的使用

    步骤一:ctrl+alt+t打开终端,打开theanorc文件

    sudo gedit ~/.theanorc

    空白文件中写入:

    [global]

    floatX=float32

    device=cpu

    步骤二:修改内容在上边基础上修改

    1、使用gpu加速时,把上边device修改,同时添加新的内容

    device=gpu

    [cuda]

    root=/usr/local/cuda-8.0

    2、使用cudnn加速时,继续添加

    [nvcc]

    fastmath =True

    optimizer_including=cudnn

    3、想要保证使用cuda运行过程中参数不随机时候添加设置

    [dnn.conv]

    algo_bwd_filter =deterministic

    algo_bwd_data =deterministic

    4、并行操作时配置添加

    默认情况下,显卡显存占满,只允许一个程序运行如下所示:初始化时调用30%的显存,可最大允许三个程序同时运行。

    [lib]

    cnmem=0.3

    ref:Ubuntu theano cuda 配置 之 theanorc文件 Mark

    相关文章

      网友评论

          本文标题:Ubuntu下theanorc文件配置

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