@Mia Hu I think your scenario needs to use a custom docker image for training. In that case, you can refer to the steps in this documentation to use this image in your build.
If an answer is helpful, please click on or upvote which might help other community members reading this thread.