caffe使用中的问题
- 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使用建议
- 最近在使用pytorch中遇到的问题,python默认开启了多线程,导致一个程序占据占据了服务器的大半资源,可通过export OMP_NUM_THREADS=1,将当前终端限制只使用单线程,该方法对pycaffe也有效。python设置可通过
import os
os.environ['OMP_NUM_THREADS'] = '1'
如果在python中使用,建议放到主函数最前边的位置。