Improve docker files and use a non-root user
This commit is contained in:
1 parent
2bd1a5b43b
commit
4041defb8d
7 files changed
+40
-35
No files matched your search
@@ -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}
|
||||
Reference in new issue
Block a user