python - theano 不能用GPU
阿神
阿神 2017-04-18 10:32:33
0
0
546

安装上CUDA8 和cudnn5之后,配置了/.theanorc文件。
在CMD窗口,nvcc -V也显示了cuda信息。
找了cuda samples测试了 result pass
按说cuda安装应该没有问题了。

但是
">>>import theano
">>>
没有出现GPU的信息。。。。
用测试代码测试了一个小程序,结果也是Used the cpu

我觉得可能还是theano的GPU配置的问题
有没有小伙伴也遇到过类似的问题??

阿神
阿神

闭关修行中......

reply all(0)
Latest Downloads
More>
Web Effects
Website Source Code
Website Materials
Front End Template