bug信息显然是提示更新theano版本,可是一看官网最新版本就是0.7.0,使用pip更新后也提示已经是最新版本。
经过检查后发现,原来theano有两个版本,一个是稳定版,通过 pip install theano
获得的就是稳定版,还有一个叫做‘bleeding-edge version’,即最新版,必须通过git更新。
命令如下pip install git+git://github.com/Theano/Theano.git --upgrade --no-deps
(–no-deps的意思是单独安装theano,各种依赖我已经装好了),注意电脑上必须先装有git.
2.内存足够却还是提示内存不足
MemoryError: Error allocating 16777216 bytes of device memory (an illegal memory access was encountered). Apply node that caused the error: GpuElemwise{mul,no_inplace}(CudaNdarrayConstant{error while transferring the value: error copying data to host}, GpuDimShuffle{x,x}.0, image_hidden_w) Inputs types: [CudaNdarrayType(float32, (True, True)), CudaNdarrayType(float32, (True, True)), CudaNdarrayType(float32, matrix)]
方法还是将theano更新到开发者版本
记录两个命令:
pip install --upgrade --no-deps git+ git://github.com/fchollet/keras.git pip install --upgrade --no-deps git+git://github.com/Theano/Theano.git
分别是更新最新版本theano和keras
调用gpu运行程序
CUDA_LAUNCH_BLOCKING=1 THEANO_FLAGS=device=gpu,floatX=float32,cuda.root=/usr/local/cuda-6.5/ python code.py
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:theano和keras使用过程中遇到的一些问题记录 - Python技术站