 
                        在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
这个问题困扰很久了,查了很多资料也没有解决,第一次上来提问,希望能得到解决,谢谢!
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号
学习是最好的投资!