diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index 7d480388..a294d2c0 100644 --- a/docker/Dockerfile.ubuntu +++ b/docker/Dockerfile.ubuntu @@ -45,7 +45,8 @@ else \ pip cache purge; \ fi -RUN echo "cache bust $(date +%Y%m%d%H%M%S)" && true +ARG CUR_TIME={cur_time} +RUN echo $CUR_TIME RUN sh /tmp/install.sh {version_args} && \ curl -fsSL https://ollama.com/install.sh | sh && \ diff --git a/docker/build_image.py b/docker/build_image.py index d614f521..5f253eae 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -160,6 +160,7 @@ class CPUImageBuilder(Builder): content = content.replace('{extra_content}', extra_content) content = content.replace('{meta_file}', meta_file) content = content.replace('{version_args}', version_args) + content = content.replace('{cur_time}', formatted_time) content = content.replace('{install_ms_deps}', 'True') content = content.replace('{torch_version}', self.args.torch_version) @@ -222,6 +223,7 @@ RUN pip install tf-keras==2.16.0 --no-dependencies && \ content = content.replace('{extra_content}', extra_content) content = content.replace('{meta_file}', meta_file) content = content.replace('{version_args}', version_args) + content = content.replace('{cur_time}', formatted_time) content = content.replace('{install_ms_deps}', 'True') content = content.replace('{torch_version}', self.args.torch_version) @@ -296,6 +298,7 @@ class LLMImageBuilder(Builder): content = content.replace('{extra_content}', extra_content) content = content.replace('{meta_file}', meta_file) content = content.replace('{version_args}', version_args) + content = content.replace('{cur_time}', formatted_time) content = content.replace('{install_ms_deps}', 'False') content = content.replace('{torch_version}', self.args.torch_version)