Caffe: Cannot create Cublas handle. Cublas won't be available


在使用Caffe生成LDMB数据过程中出现下列错误: 

  E0721 19:51:05.899822 24949 common.cpp:114] Cannot create Cublas handle. Cublas won't be available.
  E0721 19:51:05.957212 24949 common.cpp:121] Cannot create Curand generator. Curand won't be available.

奇怪的是,同样的代码,之前也使用过,没有出现这样的问题,在网上搜索发现这篇博客:

   https://xmfbit.github.io/2018/02/08/bug-pycaffe-using-cublas/

原因是第0号显卡内存已满,导致出现这种情况,通过在运行脚本前指定可见显卡,可以解决该问题:

  CUDA_VISIBLE_DEVICES=1 bash create_lmdb.sh


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM