What you would like to be added?
Always install additional python packages defined in packages_to_install with the --user argument.
Why is this needed?
When installing the packages listed in packages_to_install they are currently installed system wide with pip install <packages>.
When the training container is running as non root and the HOME location is not accessible to this user, the install fails.
By always installing with --user we will ensure the install will not fail for write permissions reasons. Users can use this in conjunction with setting a HOME environment variable.
See issues/84 for more details.
Love this feature?
Give it a 👍 We prioritize the features with most 👍