Theano 错误信息记录

最新推荐文章于 2020-05-08 15:28:41 发布

李亚超

最新推荐文章于 2020-05-08 15:28:41 发布

阅读量1.7k

点赞数

CC 4.0 BY-SA版权

分类专栏： Deep Learning Python

本文链接：https://blog.youkuaiyun.com/Harry_lyc/article/details/80720859

Deep Learning 同时被 2 个专栏收录

7 篇文章

订阅专栏

Python

3 篇文章

订阅专栏

在Ubuntu 14.0的环境下，使用CUDA 8.0和cuDNN 6.0时遇到Theano库的错误，具体表现为Theano的softmax函数报错。问题根源在于Theano要求cuDNN版本为5.1。解决方案是将cuDNN降级到5.1，目前未发现其他有效替代方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

错误信息：

Exception: ('The following error happened while compiling the node', GpuDnnSoftmax{tensor_format='bc01', mode='channel', algo='accurate'}(GpuContiguous.0), '\n', 'nvcc return status', 2, 'for cmd', 'nvcc -shared -O3 -Xlinker -rpath,/usr/local/cuda/lib64 -arch=sm_61 -m64 -Xcompiler -fno-math-errno,-Wno-unused-label,-Wno-unused-variable,-Wno-write-strings,-DCUDA_NDARRAY_CUH=c72d035fdf91890f3b36710688069b2e,-DNPY_NO_DEPRECATED_API=NPY_1_7_API_VERSION,-fPIC,-fvisibility=hidden -Xlinker -rpath,/home/ycli/.theano/compiledir_Linux-3.16--generic-x86_64-with-Ubuntu-14.04-trusty-x86_64-2.7.6-64/cuda_ndarray -I/home/ycli/.theano/compiledir_Linux-3.16--generic-x86_64-with-Ubuntu-14.04-trusty-x86_64-2.7.6-64/cuda_ndarray -I/usr/local/cuda/include -I/usr/local/lib/python2.7/dist-packages/theano/sandbox/cuda -I/usr/local/lib/python2.7/dist-packages/numpy/core/include -I/usr/include/python2.7 -I/usr/local/lib/python2.7/dist-packages/theano/gof -L/home/ycli/.theano/compiledir_Linux-3.16--generic-x86_64-with-Ubuntu-14.04-trusty-x86_64-2.7.6-64/cuda_ndarray -L/usr/lib -o /home/ycli/.theano/compiledir_Linux-3.16--generic-x86_64-with-Ubuntu-14.04-trusty-x86_64-2.7.6-64/tmpWsfM_p/4532e6b24a81e914613fd46d9f2cfae2.so mod.cu -lcudart -lcublas -lcuda_ndarray -lcudnn -lpython2.7', "[GpuDnnSoftmax{tensor_format='bc01', mode='channel', algo='accurate'}(<CudaNdarrayType(float32, (False, False, True, True))>)]")

条件：ubuntu 14.0，cuda 8.0, cudnn 6.0, theano 0.9

问题：Theano需要cudnn5.1的版本，错误出在theano.nn.softmax()函数。

问题解决：只有减低cudnn版本为5.1，好像实在找不到其他更好的办法了。