ik_llama_docker/create_new_vllm_image.sh
2026-05-16 15:02:45 -05:00

24 lines
485 B
Bash
Executable File

#!/bin/bash
export BASE_IMAGE="vllm/vllm-openai:v0.20.1-cu129"
export DATE=$(date +%Y-%m-%d)
export IMG_PRE="vllm-dw"
export IMG_NAME="$IMG_PRE:$DATE"
cat > Dockerfile << EOF
FROM $BASE_IMAGE
RUN groupadd -g 1000 vllm && useradd -u 1000 -g 1000 -m -s /bin/bash vllm
WORKDIR /home/vllm
USER vllm
EOF
docker build -t $IMG_NAME .
#Updated env for other scripts
cat > ./env/vllm_env.sh << EOF
#!/bin/bash
#updated from create_new_image_vllm.sh
export IMAGE=$IMG_NAME
EOF
rm Dockerfile