调用GPU运行CNN程序时遇到Cublas error
在Ubuntu 14.04尝试使用GPU运行mnielsen的样例代码 src/network3.py
https://github.com/mnielsen/n...
时报错Cublas error
调用代码是
import network3
from network3 import Network
from network3 import ConvPoolLayer,FullyConnectedLayer,SoftmaxLayer
traning_data,validation_data,test_data=network3.load_data_shared()
expanded_training_data,_,_=network3.load_data_shared()
mini_batch_size=10
net=Network([
ConvPoolLayer(image_shape=(mini_batch_size,1,28,28),
filter_shape=(20,1,5,5),
poolsize=(2,2)
),
ConvPoolLayer(image_shape=(mini_batch_size,20,12,12),
filter_shape=(40,20,5,5),
poolsize=(2,2)
),
FullyConnectedLayer(n_in=40*4*4,n_out=100),
FullyConnectedLayer(n_in=100,n_out=100),
SoftmaxLayer(n_in=100,n_out=10)],mini_batch_size)
net.SGD(expanded_training_data,40,mini_batch_size,0.03,validation_data, test_data)
运行结果是
RuntimeError: GpuCorrMM encountered a CUBLAS error: the library was not initialized
This could be a known bug in CUDA, please see the GpuCorrMM() documentation.
theanorc的配置是
[global]
openmp=False
floatX=float32
device=gpu
allow_input_downcast=True
[blas]
ldflags=-lopenblas
[nvcc]
flags=-LC:/home/fang/anaconda2/lib
fastmath=True
[cuda]
root=/usr/local/cuda-8.0
GPU使用的是Nvidia GeForce GTX 460
其它环境分别是cuda 8.0, theano 0.8.2, 使用的调试平台是eclipse pydev
这个问题困扰很久了,查了很多资料也没有解决,第一次上来提问,希望能得到解决,谢谢!
如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论