Reduce image size -Install torch for CPU

Author: alhendrickson

Dockerfile

@@ -11,8 +11,10 @@ ARG USE_CPU_TORCH=false
 # NOTE: Allow building without GPU so as to lower image size (disabled by default)
 RUN pip install -U pip && \
     if [ "${USE_CPU_TORCH}" = "true" ]; then \
-        pip install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu/; \
+        echo "Installing Torch with CPU index URL..." && \
+        pip install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu; \
     else \
+        echo "Installing Torch with GPU support..." && \
         pip install --no-cache-dir -r requirements.txt; \
     fi