caffe训练时总是会占用0号GPU卡

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/x_j_b/article/details/73530541

 Sometimes, we will change the default GPU 0  to others when  we are training deep learning models using Caffe, however, we also find that gpu 0 is always used (~100 MBs) 

.To solver this, we can simply change the order of the GPU setting language: 

caffe.set_mode_gpu()

caffe.set_device(2)

=====>

caffe.set_device(2)

caffe.set_mode_gpu()


展开阅读全文

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