search

dividiti / ck-caffe

Collective Knowledge workflow for Caffe to automate installation across diverse platforms and to collaboratively evaluate and optimize Caffe-based workloads across diverse hardware, software and data sets (compilers, libraries, tools, models, inputs):
http://cKnowledge.org
BSD 3-Clause "New" or "Revised" License
193 stars 40 forks source link

Add env CAFFE_COMPUTE_DEVICE_ID to caffe-classify-* to be able to select differnet gpu #46

Closed DVEfremov closed 7 years ago

DVEfremov commented 7 years ago

Add env CAFFE_COMPUTE_DEVICE_ID to caffe-classify-* to be able to select differnet gpu. It's currently hardwired to 0

DVEfremov commented 7 years ago

fixed, commits merged to master