python 2.7 - How can I make ensure caffe using GPU? -
is there way ensure caffe using gpu? compiled caffe after installing cuda driver , without cpu_only
flag in cmake , while compiling cmake logged detection of cuda 8.0.
but while train sample, doubt using gpu according nvidia-smi
result. how can ensure?
the surest way know configure solver.prototxt file. include line
solver_mode: gpu
if have specifications engine use in each layer of model, you'll want make sure refer gpu software.
Comments
Post a Comment