关于theano和keras的一些安装总结

        Keras是一个基于theano和tensorflow的库框架,它的安装要不是Using Theano backend 要不就是Using tensorflow backend 。所以一般在安装keras之前一定要装好theano或者tensorflow,至于具体选那种根据自己的实际情况在keras.json文件中配置即可。这个大家都差不多可以搜索一下即可配置好。
        但是在theano.rc或者tensorflow.rc上的配置往往决定你的keras框架能不能用起来,一个非常终于的点是device的选择和cuda地址的写入。笔者开始就是这几个地方出问题产生了各种虚头巴脑的错误,花了很多时间都是一些看起来没有任何联系的问题。下面贴出我的配置:
[global]
openmp=False
device = cuda0
optimizer=fast_compile    
floatX = float32 
root=/usr/local/cuda-8.0
allow_input_downcast=True 
[lib]
cnmem = 0.8
[blas]
ldflags= -lopenblas
[nvcc]
fastmath = True
[cuda]
root=/usr/local/cuda-8.0
       注意:cnmem是GPU使用在这个终端或者这次运行中所占的比例。所以在设置的时候不能太偏执。我的一些代码编译提示如下图:
       注意:device的配置可以是GPU0或者cuda0等,大家可以多尝试,如果说pygpu或者pyarry等等有问题都是device没有配置好。   
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值