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 \