diff --git a/ge/init/gelib.cc b/ge/init/gelib.cc index 20859c07..03028e53 100755 --- a/ge/init/gelib.cc +++ b/ge/init/gelib.cc @@ -146,7 +146,7 @@ Status GELib::InnerInitialize(const map &options) { } if (is_train_mode_ || (options_.device_id != kDefaultDeviceIdForInfer)) { - GE_CHK_RT_RET(rtSetDevice(options.device_id)); + GE_CHK_RT_RET(rtSetDevice(options_.device_id)); } GELOGI("sessionManager initial.");