Fix apex installation path in Dockerfile (#11596)

* empty commit * Specify apex installation target directory * pip install --user
Lightning-AI · Jan 28, 2022 · 86b177e · 86b177e
1 parent 11e1e3e
commit 86b177e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 \