50 lines
1.3 KiB
Bash
Executable File
50 lines
1.3 KiB
Bash
Executable File
#!/bin/sh
|
|
set -e
|
|
|
|
IMAGE="${DNC_IMAGE:-ghcr.io/lstmnemodel/dnc:latest}"
|
|
|
|
GPU=none
|
|
if command -v nvidia-smi >/dev/null 2>&1; then
|
|
GPU=nvidia
|
|
fi
|
|
|
|
DRI=""
|
|
[ -d /dev/dri ] && DRI=1
|
|
|
|
KFD=""
|
|
[ -e /dev/kfd ] && KFD=1
|
|
|
|
DOCKER_SOCK="/var/run/docker.sock"
|
|
|
|
ROOTLESS=$(docker info --format '{{.ClientInfo.Context}}' 2>/dev/null | grep -ci rootless || true)
|
|
if [ "$ROOTLESS" -gt 0 ] 2>/dev/null ; then
|
|
echo "ERROR: dnc requires rootful Docker. Rootless Docker is not supported." >&2
|
|
exit 1
|
|
fi
|
|
|
|
DNC_CACHE="${XDG_CACHE_HOME:-$HOME/.cache}/dnc"
|
|
mkdir -p "$DNC_CACHE"
|
|
|
|
TTY_FLAG=""
|
|
[ -t 0 ] && TTY_FLAG="-t"
|
|
|
|
exec docker run --rm -i ${TTY_FLAG} \
|
|
-v "$DOCKER_SOCK:/var/run/docker.sock:ro" \
|
|
-v "$PWD:$PWD" -w "$PWD" \
|
|
-v "$DNC_CACHE:/opt/dnc/host:rw" \
|
|
-e "DNC_HOST_GPU=$GPU" \
|
|
-e "DNC_HOST_DRI=$DRI" \
|
|
-e "DNC_HOST_KFD=$KFD" \
|
|
-e "DNC_HOST_UID=$(id -u)" \
|
|
-e "DNC_HOST_GID=$(id -g)" \
|
|
-e "DNC_HOST_USER=$USER" \
|
|
-e "DNC_HOST_HOME=$HOME" \
|
|
-e "DNC_HOST_SHELL=$SHELL" \
|
|
-e "DNC_HOST_GIDS=$(id -G)" \
|
|
-e "DNC_CACHE_HOST=$DNC_CACHE" \
|
|
-e DISPLAY -e WAYLAND_DISPLAY -e XAUTHORITY \
|
|
-e DBUS_SESSION_BUS_ADDRESS -e XDG_RUNTIME_DIR \
|
|
-e PULSE_SERVER -e PIPEWIRE_RUNTIME_DIR \
|
|
-e SSH_AUTH_SOCK -e TERM -e LANG \
|
|
"$IMAGE" "$@"
|