Ubuntu下theanorc文件配置

配置这个文件的目的是,可以切换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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容