Update conda instead of ignoring installed

Signed-off-by: Fredrik Enestad <fredrik@enestad.com>
NVIDIA · May 24, 2023 · d8676fd · d8676fd
1 parent 698804d
commit d8676fd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docker/Dockerfile.torch b/docker/Dockerfile.torch
@@ -44,8 +44,9 @@ RUN git submodule update --init --recursive && \
 # RUN pip uninstall -y apex && \
 #     pip install git+https://github.com/NVIDIA/apex --global-option="--cpp_ext" --global-option="--cuda_ext" --global-option="--fast_layer_norm" && \
 #     pip install 3rdparty/NeMo[nlp]
+RUN conda update --all
 RUN pip install --upgrade --no-cache-dir pip \
- && pip install --no-cache-dir 3rdparty/NeMo[nlp] --ignore-installed
+ && pip install --no-cache-dir 3rdparty/NeMo[nlp]
 
 ENV NCCL_LAUNCH_MODE=GROUP
 ARG SM=80