关于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没有配置好。   

猜你喜欢

转载自blog.csdn.net/jacktangyao/article/details/77417143
今日推荐