mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-03-14 20:27:24 +00:00
92286e1d4a
* devcontainer: replace VAULT_HOST with AWS_ROLE_ARN Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com> * Update devcontainers base image to support AWS_ROLE_ARN Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com> * Bump cuda latest version to 12.6 Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com> * Replace ubuntu18.04 with ubuntu20.04 Ubuntu 18.04 is not supported anymore Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com> * Use DOOD stategy to keep supporting ubuntu18.04 See https://github.com/NVIDIA/cccl/pull/1779 Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com> --------- Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com>
50 lines
2.3 KiB
Bash
Executable File
50 lines
2.3 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# Maybe change the UID/GID of the container's non-root user to match the host's UID/GID
|
|
|
|
: "${REMOTE_USER:="coder"}";
|
|
: "${OLD_UID:=}";
|
|
: "${OLD_GID:=}";
|
|
: "${NEW_UID:=}";
|
|
: "${NEW_GID:=}";
|
|
|
|
eval "$(sed -n "s/${REMOTE_USER}:[^:]*:\([^:]*\):\([^:]*\):[^:]*:\([^:]*\).*/OLD_UID=\1;OLD_GID=\2;HOME_FOLDER=\3/p" /etc/passwd)";
|
|
eval "$(sed -n "s/\([^:]*\):[^:]*:${NEW_UID}:.*/EXISTING_USER=\1/p" /etc/passwd)";
|
|
eval "$(sed -n "s/\([^:]*\):[^:]*:${NEW_GID}:.*/EXISTING_GROUP=\1/p" /etc/group)";
|
|
|
|
if [ -z "$OLD_UID" ]; then
|
|
echo "Remote user not found in /etc/passwd ($REMOTE_USER).";
|
|
exec "$(pwd)/.devcontainer/nvbench-entrypoint.sh" "$@";
|
|
elif [ "$OLD_UID" = "$NEW_UID" ] && [ "$OLD_GID" = "$NEW_GID" ]; then
|
|
echo "UIDs and GIDs are the same ($NEW_UID:$NEW_GID).";
|
|
exec "$(pwd)/.devcontainer/nvbench-entrypoint.sh" "$@";
|
|
elif [ "$OLD_UID" != "$NEW_UID" ] && [ -n "$EXISTING_USER" ]; then
|
|
echo "User with UID exists ($EXISTING_USER=$NEW_UID).";
|
|
exec "$(pwd)/.devcontainer/nvbench-entrypoint.sh" "$@";
|
|
else
|
|
if [ "$OLD_GID" != "$NEW_GID" ] && [ -n "$EXISTING_GROUP" ]; then
|
|
echo "Group with GID exists ($EXISTING_GROUP=$NEW_GID).";
|
|
NEW_GID="$OLD_GID";
|
|
fi
|
|
echo "Updating UID:GID from $OLD_UID:$OLD_GID to $NEW_UID:$NEW_GID.";
|
|
sed -i -e "s/\(${REMOTE_USER}:[^:]*:\)[^:]*:[^:]*/\1${NEW_UID}:${NEW_GID}/" /etc/passwd;
|
|
if [ "$OLD_GID" != "$NEW_GID" ]; then
|
|
sed -i -e "s/\([^:]*:[^:]*:\)${OLD_GID}:/\1${NEW_GID}:/" /etc/group;
|
|
fi
|
|
|
|
# Fast parallel `chown -R`
|
|
find "$HOME_FOLDER/" -not -user "$REMOTE_USER" -print0 \
|
|
| xargs -0 -r -n1 -P"$(nproc --all)" chown "$NEW_UID:$NEW_GID"
|
|
|
|
# Run the container command as $REMOTE_USER, preserving the container startup environment.
|
|
#
|
|
# We cannot use `su -w` because that's not supported by the `su` in Ubuntu18.04, so we reset the following
|
|
# environment variables to the expected values, then pass through everything else from the startup environment.
|
|
export HOME="$HOME_FOLDER";
|
|
export XDG_CACHE_HOME="$HOME_FOLDER/.cache";
|
|
export XDG_CONFIG_HOME="$HOME_FOLDER/.config";
|
|
export XDG_STATE_HOME="$HOME_FOLDER/.local/state";
|
|
export PYTHONHISTFILE="$HOME_FOLDER/.local/state/.python_history";
|
|
exec su -p "$REMOTE_USER" -- "$(pwd)/.devcontainer/nvbench-entrypoint.sh" "$@";
|
|
fi
|