标签:
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.
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
标签:
原文地址:http://www.cnblogs.com/morningsky/p/5448992.html