caffe 指定使用的GPU

版权声明: https://blog.csdn.net/b876144622/article/details/79972638

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

阅读更多

扫码向博主提问

keep_forward

非学,无以致疑;非问,无以广识
去开通我的Chat快问
想对作者说点什么? 我来说一句

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