调用GPU运行CNN程序时遇到Cublas error

发布于 2022-09-04 09:23:36 字数 1751 浏览 13 评论 0

在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 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文