#!/bin/bash
export BASE_IMAGE="nvcr.io/nvidia/cuda:13.2.0-cudnn-devel-ubuntu24.04"

docker run --name llama-dw-$(date +%Y-%m-%d) --gpus=all $BASE_IMAGE /bin/bash -c \
'apt-get update
apt-get --yes upgrade
apt-get --yes install build-essential git libcurl4-openssl-dev curl libgomp1 cmake
git clone https://github.com/ggml-org/llama.cpp
cd llama.cpp
cmake -B build -DGGML_NATIVE=ON -DGGML_CUDA=ON
cmake --build build --config Release -j$(nproc)'
docker commit llama-dw-$(date +%Y-%m-%d) llama-dw:$(date +%Y-%m-%d)
docker rm llama-dw-$(date +%Y-%m-%d)

#Updated env for other scripts
IMAGE=llama-dw:$(date +%Y-%m-%d)

cat > ./env/llama_env.sh << EOF
#!/bin/bash
#updated from create_new_image.sh
export IMAGE=$IMAGE
EOF