File: set_device.m

package info (click to toggle)
caffe 1.0.0%2Bgit20180821.99bd997-8
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 16,288 kB
  • sloc: cpp: 61,586; python: 5,783; makefile: 599; sh: 559
file content (11 lines) | stat: -rw-r--r-- 250 bytes parent folder | download | duplicates (5) 
1
2
3
4
5
6
7
8
9
10
11
 
function set_device(device_id)
% set_device(device_id)
%   set Caffe's GPU device ID

CHECK(isscalar(device_id) && device_id >= 0, ...
  'device_id must be non-negative integer');
device_id = double(device_id);

caffe_('set_device', device_id);

end