caffe 指定使用的GPU

caffe若在GPU模式下,会默认是使用0号GPU,且默认只使用需要的GPU内存。(这一点和tensorflow不一样,tensorflow会默认使用所有的GPU,以及所有的GPU内存),如果0号GPU内存满了,即使其它GPU空闲,caffe也不会使用其它GPU


在运行caffe程序前,可以先用nvidia-smi命令查看哪些GPU空闲,然后指定某个GPU使用,比如

train.sh -gpu 2  #使用2号GPU


train.sh -gpu all #使用所有的GPU


train.sh -gpu 1,2,3 #使用1,2,3号GPU

阅读更多
版权声明: https://blog.csdn.net/b876144622/article/details/79972638
文章标签: caffe gpu 指定
个人分类: caffe
上一篇putty配色设置
下一篇深度学习中的softmax和cross entropy
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

caffe 指定使用的GPU

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭