summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: 49c5345)
raw | patch | inline | side by side (parent: 49c5345)
author | Yangqing Jia <jiayq84@gmail.com> | |
Tue, 26 Nov 2013 18:59:06 +0000 (10:59 -0800) | ||
committer | Yangqing Jia <jiayq84@gmail.com> | |
Tue, 26 Nov 2013 18:59:06 +0000 (10:59 -0800) |
examples/net_speed_benchmark.cpp | patch | blob | history |
index 7a0b375ef8ea5ae0ce35365b55b63b1fa7c5e39c..a287591cba9bb12c1987a6bdec42f8da5272d798 100644 (file)
using namespace caffe;
int main(int argc, char** argv) {
- cudaSetDevice(1);
+ cudaSetDevice(0);
Caffe::set_mode(Caffe::GPU);
Caffe::set_phase(Caffe::TRAIN);
int repeat = 100;