keep forward, go, go, go

主要用于自己备忘,写的不太好,请轻拍,有疑问请留言,共同讨论交流^_^

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

阅读更多

扫码向博主提问

去开通我的Chat快问

b876144622

非学,无以致疑;非问,无以广识
版权声明: https://blog.csdn.net/b876144622/article/details/79972638
文章标签: caffe gpu 指定
个人分类: caffe
上一篇putty配色设置
下一篇python 数组反转
想对作者说点什么? 我来说一句

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

关闭
关闭