Merge: Update dllogger installation

Author: nv-kkudrynski

PyTorch/Forecasting/TFT/Dockerfile

@@ -24,11 +24,12 @@ RUN apt-get update && apt-get install -y \
 
 WORKDIR /workspace
 
+RUN pip install --no-cache-dir nvidia-pyindex
+
 COPY requirements.txt .
 
 RUN pip uninstall -y typing && \
-    pip install --no-cache-dir --ignore-installed -r requirements.txt && \
-    pip install --no-cache-dir -e git+https://github.com/NVIDIA/dllogger#egg=dllogger
+    pip install --no-cache-dir --ignore-installed -r requirements.txt
 
 COPY . .
 

PyTorch/Forecasting/TFT/requirements.txt

@@ -1,2 +1,3 @@
+nvidia-dllogger==0.1.0
 pandas==1.3.4
 pynvml==11.0.0

TensorFlow/Recommendation/WideAndDeep/Dockerfile

@@ -39,10 +39,11 @@ ENV SPARK_HOME /usr/local/spark
 ENV PYTHONPATH $SPARK_HOME/python:$SPARK_HOME/python/lib/py4j-0.10.7-src.zip
 ENV SPARK_OPTS --driver-java-options=-Xms1024M --driver-java-options=-Xmx4096M --driver-java-options=-Dlog4j.logLevel=info
 
-RUN pip install pyspark==2.3.1
-RUN pip install --no-deps tensorflow-transform==0.24.1 apache-beam==2.14 tensorflow-metadata==0.14.0 pydot dill
-RUN pip install ipdb
-RUN pip install -e git://github.com/NVIDIA/dllogger#egg=dllogger
+RUN pip install --no-cache-dir nvidia-pyindex
+
+COPY requirements* ./
+RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir --no-deps -r requirements-no-deps.txt
 
 WORKDIR  /wd
 COPY . .

TensorFlow/Recommendation/WideAndDeep/requirements-no-deps.txt

@@ -0,0 +1,5 @@
+tensorflow-transform==0.24.1
+apache-beam==2.14
+tensorflow-metadata==0.14.0
+pydot
+dill

TensorFlow/Recommendation/WideAndDeep/requirements.txt

@@ -0,0 +1,3 @@
+nvidia-dllogger==0.1.0
+ipdb==0.13.9
+pyspark==2.3.1

TensorFlow2/Recommendation/WideAndDeep/Dockerfile

@@ -20,10 +20,9 @@ ENV HOROVOD_CYCLE_TIME=0.1
 ENV HOROVOD_FUSION_THRESHOLD=67108864
 ENV HOROVOD_NUM_STREAMS=2
 
-
-USER root
-
-RUN pip install --no-cache-dir -e git+https://github.com/NVIDIA/dllogger#egg=dllogger
+RUN pip install --no-cache-dir nvidia-pyindex
+ADD requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
 WORKDIR  /wd
 

TensorFlow2/Recommendation/WideAndDeep/requirements.txt

@@ -0,0 +1 @@
+nvidia-dllogger==0.1.0