Keep nvcc for MSDeepSpeed Docker image

pytorch · Feb 27, 2021 · d2bdf83 · d2bdf83
1 parent e6950b1
commit d2bdf83
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/docker/msdp/Dockerfile.msdp-apex b/docker/msdp/Dockerfile.msdp-apex
@@ -67,10 +67,7 @@ RUN pip install --upgrade --no-cache-dir pytorch-ignite \
 # replace pillow with pillow-simd
 RUN apt-get update && apt-get -y install --no-install-recommends g++ && \
     pip uninstall -y pillow && \
-    CC="cc -mavx2" pip install --upgrade --no-cache-dir --force-reinstall pillow-simd && \
-    apt-get remove -y g++ && \
-    apt-get autoremove -y && \
-    rm -rf /var/lib/apt/lists/*
+    CC="cc -mavx2" pip install --upgrade --no-cache-dir --force-reinstall pillow-simd
 
 # Checkout Ignite examples only
 RUN mkdir -p pytorch-ignite-examples && \