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