def METHOD_NAME(self):
    X, y = build_dataset()
    dtrain = xgb.DMatrix(X, y)
    bst = xgb.train(
        {"tree_method": "gpu_hist", "gpu_id": 1}, dtrain, num_boost_round=6
    )
    model_path = "model.pkl"
    save_pickle(bst, model_path)
    cuda_environment = {"CUDA_VISIBLE_DEVICES": "0"}
    env = os.environ.copy()
    env.update(cuda_environment)
    args = self.args_template.copy()
    args.append(
        "./tests/python-gpu/" "load_pickle.py::TestLoadPickle::test_wrap_gpu_id"
    )
    status = subprocess.call(args, env=env)
    assert status == 0
    os.remove(model_path)