Improve docker files and use a non-root user

This commit is contained in:
ekoc committed 2025-03-14 14:14:18 +01:00
1 parent 2bd1a5b43b
commit 4041defb8d
7 files changed
+40 -35

No files matched your search

+2 -2
View File
@@ -7,6 +7,7 @@ cd $SCRIPT_DIR
DOCKER_TAG=latest
DOCKER_REPOSITORY=erew123
ALLTALK_DIR=/home/alltalk
# Parse arguments
while [ "$#" -gt 0 ]; do
@@ -67,6 +68,5 @@ docker run \
-it \
--gpus=all \
--name deepspeed \
--build-arg DOCKER_TAG=$DOCKER_TAG \
-v $SCRIPT_DIR/build:/deepspeed \
-v $SCRIPT_DIR/build:${ALLTALK_DIR}/build \
${DOCKER_REPOSITORY}alltalk_deepspeed:${DOCKER_TAG}