美文网首页
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