В системе с 2 NVidia 1080 ti (Tensorflow 1.12 / Cuda 10), вызывающей tennflowflow :: graph :: SetDefaultDevice для выбора конкретного 1080, игнорируется, и график всегда выполняется на gpu: 0 (согласно nvidia-smi)
, называемый tennsflow :: graph :: SetDefaultDevice, как описано в документации в Интернете.
if (check_status (ReadBinaryProto (tf :: Env :: Default (), graphFilePath, & graph_))){
tf::SessionOptions options;
options.config.set_log_device_placement(true);
options.config.set_allow_soft_placement(true);
if (check_status(tf::NewSession(tf::SessionOptions(), &session_)))
{
if (check_status(session_->Create(graph_)))
{
tensorflow::graph::SetDefaultDevice("/gpu:1", &(graph_));
success = true;
}
else
{
session_->Close();
session_= nullptr;
}
}
}
Ожидаемое поведение состоит в том, что график будет выполняться на gpu: 1, но согласно nvidia-smi график всегда выполняется на gpu: 0.