您的位置:首页 > Web前端

在使用caffe的时候,如何设置GPU的ID号

2015-01-19 15:16 405 查看
有两种情况设置GPU的ID号:

1、如果使用caffe的源码(即非Python和matlab接口)

在脚本文件中加入参数: --gpu=ID; 例如 "--gpu=1",如果不设置默认使用过的ID=0,注意这个ID数是大于等于0的数,如0,1,2.....

2、如果使用matlab的接口的时候,有两种方法

A、使用matlab自带的命令,g=gpuDevice(ID);例如,g=gpuDevice(1),注意这个ID数是大于0的数,1,2.....

B、在初始化caffe的时候,进行设置,caffe('set_device',ID); 例如 caffe('set_device',
1),注意这个ID数是大于等于0的数,如0,1,2.....
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: