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