From 66a9a0a72ec4e76f542a84a96900dbb8155179b6 Mon Sep 17 00:00:00 2001 From: c-bata Date: Wed, 11 Oct 2023 13:56:43 +0900 Subject: [PATCH] Use CPU version of PyTorch in Docker image --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 3e8b3e3a30d..e3346b07f27 100644 --- a/Dockerfile +++ b/Dockerfile @@ -16,7 +16,7 @@ COPY . . ARG BUILD_TYPE='dev' RUN if [ "${BUILD_TYPE}" = "dev" ]; then \ - pip install ${PIP_OPTIONS} -e '.[benchmark, checking, document, integration, optional, test]' -f https://download.pytorch.org/whl/torch_stable.html; \ + pip install ${PIP_OPTIONS} -e '.[benchmark, checking, document, integration, optional, test]' --extra-index-url https://download.pytorch.org/whl/cpu; \ else \ pip install ${PIP_OPTIONS} -e .; \ fi \