update error message for learning without GPU module
@@ -806,7 +806,7 @@ void TrainModel(const NJson::TJsonValue& plainJsonParams,
if (isGpuDeviceType && TTrainerFactory::Has(ETaskType::GPU)) {
modelTrainerHolder = TTrainerFactory::Construct(ETaskType::GPU);
} else {
- CB_ENSURE(!isGpuDeviceType, "Can't load GPU learning library. Module was not compiled or CUDA version/driver is incompatible with package");
+ CB_ENSURE(!isGpuDeviceType, "Can't load GPU learning library. Module was not compiled or driver is incompatible with package. Please install latest NVDIA driver and check again");
modelTrainerHolder = TTrainerFactory::Construct(ETaskType::CPU);
}
modelTrainerHolder->TrainModel(trainOptions, outputOptions, objectiveDescriptor, evalMetricDescriptor, pools, modelPtr, evalResultPtrs);
