From 86b177ebe5427725b35fde1a8808a7b59b8a277a Mon Sep 17 00:00:00 2001 From: Aki Nitta Date: Fri, 28 Jan 2022 10:14:16 +0900 Subject: [PATCH] Fix `apex` installation path in Dockerfile (#11596) * empty commit * Specify apex installation target directory * pip install --user --- dockers/base-cuda/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dockers/base-cuda/Dockerfile b/dockers/base-cuda/Dockerfile index d70761cbdd37a..a73a5b533b78c 100644 --- a/dockers/base-cuda/Dockerfile +++ b/dockers/base-cuda/Dockerfile @@ -103,7 +103,7 @@ RUN \ RUN \ # install NVIDIA apex - pip install --no-cache-dir --global-option="--cuda_ext" https://github.com/NVIDIA/apex/archive/refs/heads/master.zip && \ + pip install --user --no-cache-dir --global-option="--cuda_ext" https://github.com/NVIDIA/apex/archive/refs/heads/master.zip && \ python -c "from apex import amp" RUN \