Estimator for DeepRec is based on TensorFlow Estimator, which is high-level TensorFlow API that greatly simplifies machine learning programming. Estimators encapsulate training, evaluation, prediction, and exporting for your model.
New features in DeepRec Estimator:
- Support GRPC++ for large scale distributed training in parameter server mode.
- Support StarServer for large scale distributed training in parameter server mode.
CPU Dev Docker
| GCC Version | Python Version | IMAGE |
|---|---|---|
| 9.4.0 | 3.8.10 | alideeprec/deeprec-build:deeprec-dev-cpu-py38-ubuntu20.04 |
GPU(cuda11.6) Dev Docker
| GCC Version | Python Version | CUDA VERSION | IMAGE |
|---|---|---|---|
| 9.4.0 | 3.8.10 | CUDA 11.6.2 | alideeprec/deeprec-build:deeprec-dev-gpu-py38-cu116-ubuntu20.04 |
Develop Branch：master, Latest Release Branch: deeprec2302
Build Package Builder
bazel build //tensorflow_estimator/tools/pip_package:build_pip_packageBuild Package
bazel-bin/tensorflow_estimator/tools/pip_package/build_pip_package /tmp/estimator_whlInstalling DeepRec will install the native tensorflow-estimator by default, please reinstall the compiled Estimator.
See TensorFlow Estimator getting started guide for an introduction to the Estimator APIs.