From 2e35bb12bc79ab05c33299622d481a5148963cc5 Mon Sep 17 00:00:00 2001 From: Rob Ballantyne Date: Fri, 14 Jun 2024 18:41:40 +0100 Subject: [PATCH] fix function name --- build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh b/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh index adb93d8..40d6a8b 100755 --- a/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh +++ b/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh @@ -5,7 +5,7 @@ build_cpu_main() { build_common_run_tests } -build_cpu_install_deps() { +build_cpu_install_torch() { "$PYTHON_VENV_PIP" install --no-cache-dir \ torch==${PYTORCH_VERSION} \ torchvision \