Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly
Appearance settings

Commit 22a214e

Browse filesBrowse files
nv-dmajchrowskinv-dmajchrowski
committed
Update dllogger installation
1 parent 16c0b93 commit 22a214e
Copy full SHA for 22a214e

File tree

Expand file treeCollapse file tree

7 files changed

+21
-10
lines changed
Open diff view settings
Filter options
Expand file treeCollapse file tree

7 files changed

+21
-10
lines changed
Open diff view settings
Collapse file

‎PyTorch/Forecasting/TFT/Dockerfile‎

Copy file name to clipboardExpand all lines: PyTorch/Forecasting/TFT/Dockerfile
+3-2Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,12 @@ RUN apt-get update && apt-get install -y \
2424

2525
WORKDIR /workspace
2626

27+
RUN pip install --no-cache-dir nvidia-pyindex
28+
2729
COPY requirements.txt .
2830

2931
RUN pip uninstall -y typing && \
30-
pip install --no-cache-dir --ignore-installed -r requirements.txt && \
31-
pip install --no-cache-dir -e git+https://github.com/NVIDIA/dllogger#egg=dllogger
32+
pip install --no-cache-dir --ignore-installed -r requirements.txt
3233

3334
COPY . .
3435

Collapse file

‎PyTorch/Forecasting/TFT/requirements.txt‎

Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
1+
nvidia-dllogger==0.1.0
12
pandas==1.3.4
23
pynvml==11.0.0
Collapse file

‎TensorFlow/Recommendation/WideAndDeep/Dockerfile‎

Copy file name to clipboardExpand all lines: TensorFlow/Recommendation/WideAndDeep/Dockerfile
+5-4Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -39,10 +39,11 @@ ENV SPARK_HOME /usr/local/spark
3939
ENV PYTHONPATH $SPARK_HOME/python:$SPARK_HOME/python/lib/py4j-0.10.7-src.zip
4040
ENV SPARK_OPTS --driver-java-options=-Xms1024M --driver-java-options=-Xmx4096M --driver-java-options=-Dlog4j.logLevel=info
4141

42-
RUN pip install pyspark==2.3.1
43-
RUN pip install --no-deps tensorflow-transform==0.24.1 apache-beam==2.14 tensorflow-metadata==0.14.0 pydot dill
44-
RUN pip install ipdb
45-
RUN pip install -e git://github.com/NVIDIA/dllogger#egg=dllogger
42+
RUN pip install --no-cache-dir nvidia-pyindex
43+
44+
COPY requirements* ./
45+
RUN pip install --no-cache-dir -r requirements.txt
46+
RUN pip install --no-cache-dir --no-deps -r requirements-no-deps.txt
4647

4748
WORKDIR /wd
4849
COPY . .
Collapse file

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

Copy file name to clipboard
+5Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
tensorflow-transform==0.24.1
2+
apache-beam==2.14
3+
tensorflow-metadata==0.14.0
4+
pydot
5+
dill
Collapse file

‎TensorFlow/Recommendation/WideAndDeep/requirements.txt‎

Copy file name to clipboard
+3Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
nvidia-dllogger==0.1.0
2+
ipdb==0.13.9
3+
pyspark==2.3.1
Collapse file

‎TensorFlow2/Recommendation/WideAndDeep/Dockerfile‎

Copy file name to clipboardExpand all lines: TensorFlow2/Recommendation/WideAndDeep/Dockerfile
+3-4Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,9 @@ ENV HOROVOD_CYCLE_TIME=0.1
2020
ENV HOROVOD_FUSION_THRESHOLD=67108864
2121
ENV HOROVOD_NUM_STREAMS=2
2222

23-
24-
USER root
25-
26-
RUN pip install --no-cache-dir -e git+https://github.com/NVIDIA/dllogger#egg=dllogger
23+
RUN pip install --no-cache-dir nvidia-pyindex
24+
ADD requirements.txt .
25+
RUN pip install --no-cache-dir -r requirements.txt
2726

2827
WORKDIR /wd
2928

Collapse file

‎TensorFlow2/Recommendation/WideAndDeep/requirements.txt‎

Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
nvidia-dllogger==0.1.0

0 commit comments

Comments
0 (0)
Morty Proxy This is a proxified and sanitized view of the page, visit original site.