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

docker run --name ik_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/ikawrakow/ik_llama.cpp
cd ik_llama.cpp
cmake -B build -DGGML_NATIVE=ON -DGGML_CUDA=ON
cmake --build build --config Release -j$(nproc)'
docker commit ik_llama_dw_$(date +%Y-%m-%d) ik_llama_dw:$(date +%Y-%m-%d)
docker rm ik_llama_dw_$(date +%Y-%m-%d)

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

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