-
Notifications
You must be signed in to change notification settings - Fork 118
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Ready to be reviewed] Support Customized HCTR Repo in the unified containers #85
Conversation
refine dockerfile
replace DEV_MODE with HUGECTR_DEV_MODE
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We disabled "ARROW_ORC" temporally in this MR because of some related errors in the nightly unified container building. Per @shijieliu those errors also exist in the main branch on NVIDIA-Merlin and he has communicated with @albert17. |
@zehuanw Please resolve conflicts since containers have been updated to 22.01 release |
update to 21.12. resolve confilcts
add hdfs dependency
Design point:
Arguments supported:
HugeCTR will not be installed when HUGECTR_DEV_MODE=true
ARG HUGECTR_DEV_MODE=false
Arguments "_XXXX" are only valid when $HUGECTR_DEV_MODE==false
ARG _HUGECTR_BRANCH=master
ARG _HUGECTR_REPO="github.com/NVIDIA-Merlin/HugeCTR.git"
ARG _HUGECTR_BACKEND_BRANCH=main
ARG _HUGECTR_BACKEND_REPO="github.com/triton-inference-server/hugectr_backend"
ARG _CI_JOB_TOKEN=""