在使用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