caffe使用中遇到的问题与建议

caffe使用中的问题

1. Cannot create Cublas handle. Cublas won’t be available. Cannot create Curand generator. Curand won’t be available.
   问题描述：该问题是在用python初始化caffe模型阶段出现的。网上有很多相同问题，但原因似乎跟我不一样，有的人说是显卡驱动版本太高，而我不是，我的是因为GPU0可用显存不足，pycaffe使用时会默认在0号GPU占据100多M的显存。
   解决：用设置环境变量的方式，使得程序只可见部分显卡，如

export CUDA_VISIBLE_DEVICES=1  

或者在python中

import os 
os.environ['CUDA_VISIBLE_DEVICES'] = '1'

如果在python中使用，建议放到主函数最前边的位置。
注意，不仅可设置单GPU，还可以多卡，如

export CUDA_VISIBLE_DEVICES=1,2

---

caffe使用建议

1. 最近在使用pytorch中遇到的问题，python默认开启了多线程，导致一个程序占据占据了服务器的大半资源，可通过export OMP_NUM_THREADS=1，将当前终端限制只使用单线程，该方法对pycaffe也有效。python设置可通过

import os 
os.environ['OMP_NUM_THREADS'] = '1'

如果在python中使用，建议放到主函数最前边的位置。