Skip to content

Commit 3a8068b

Browse files
shakandrewnv-kkudrynski
authored andcommitted
Change DLLogger installation source
1 parent 589604d commit 3a8068b

File tree

25 files changed

+18
-30
lines changed

25 files changed

+18
-30
lines changed
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
git+git://github.com/NVIDIA/dllogger.git@26a0f8f1958de2c0c460925ff6102a4d2486d6cc#egg=dllogger
1+
git+https://github.com/NVIDIA/dllogger@v0.1.0#egg=dllogger

PyTorch/Classification/ConvNets/Dockerfile

-1
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@ FROM ${FROM_IMAGE_NAME}
33

44
ADD requirements.txt /workspace/
55
WORKDIR /workspace/
6-
RUN pip install nvidia-pyindex
76
RUN pip install --no-cache-dir -r requirements.txt
87
ADD . /workspace/rn50
98
WORKDIR /workspace/rn50
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
nvidia-dllogger==0.1.0
1+
git+https://github.com/NVIDIA/dllogger@v1.0.0#egg=dllogger
22
pynvml==11.0.0

PyTorch/Classification/ConvNets/triton/resnet50/Dockerfile

-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ ENV LD_LIBRARY_PATH /workspace/install/lib:${LD_LIBRARY_PATH}
1818
ENV PYTHONPATH /workspace
1919
WORKDIR /workspace
2020

21-
RUN pip install nvidia-pyindex
2221
ADD requirements.txt /workspace/requirements.txt
2322
ADD triton/requirements.txt /workspace/triton/requirements.txt
2423
RUN pip install -r /workspace/requirements.txt

PyTorch/Forecasting/TFT/Dockerfile

-2
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,6 @@ RUN apt-get update && apt-get install -y \
2424

2525
WORKDIR /workspace
2626

27-
RUN pip install --no-cache-dir nvidia-pyindex
28-
2927
COPY requirements.txt .
3028

3129
RUN pip uninstall -y typing && \
+1-1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
1-
nvidia-dllogger==0.1.0
1+
git+https://github.com/NVIDIA/dllogger@v1.0.0#egg=dllogger
22
pandas==1.3.4
33
pynvml==11.0.0

PyTorch/LanguageModeling/Transformer-XL/Dockerfile

+1-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ FROM ${FROM_IMAGE_NAME}
1818
ENV LANG C.UTF-8
1919
ENV LC_ALL C.UTF-8
2020

21-
RUN pip install --global-option="--cpp_ext" --global-option="--cuda_ext" git+git://github.com/NVIDIA/apex.git#egg=apex
21+
RUN pip install --global-option="--cpp_ext" --global-option="--cuda_ext" git+https://github.com/NVIDIA/apex.git#egg=apex
2222

2323
WORKDIR /workspace/transformer-xl/pytorch
2424

PyTorch/Segmentation/nnUNet/Dockerfile

-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:21.11-py3
22
FROM ${FROM_IMAGE_NAME}
33

4-
RUN pip install nvidia-pyindex
54
ADD ./requirements.txt .
65
RUN pip install --disable-pip-version-check -r requirements.txt
76
RUN pip install monai==0.8.0 --no-dependencies

PyTorch/Segmentation/nnUNet/Dockerfile-Triton

-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:21.11-py3
22
FROM ${FROM_IMAGE_NAME}
33

4-
RUN pip install nvidia-pyindex
54
ADD ./triton/requirements.txt .
65
RUN pip install --disable-pip-version-check -r requirements.txt
76
RUN apt-get update && apt-get install -y libb64-dev libb64-0d

PyTorch/SpeechRecognition/Jasper/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,4 @@ pyyaml>=5.4
77
soundfile
88
sox==1.4.1
99
tqdm==4.53.0
10-
git+git://github.com/NVIDIA/dllogger.git@26a0f8f1958de2c0c460925ff6102a4d2486d6cc#egg=dllogger
10+
git+https://github.com/NVIDIA/dllogger@v0.1.0#egg=dllogger

PyTorch/SpeechRecognition/QuartzNet/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,4 @@ pyyaml>=5.4
77
soundfile
88
sox==1.4.1
99
tqdm==4.53.0
10-
git+git://github.com/NVIDIA/dllogger.git@26a0f8f1958de2c0c460925ff6102a4d2486d6cc#egg=dllogger
10+
git+https://github.com/NVIDIA/dllogger@v0.1.0#egg=dllogger

PyTorch/SpeechSynthesis/FastPitch/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,4 @@ inflect
44
librosa==0.8.0
55
scipy
66
tensorboardX==2.0
7-
git+git://github.com/NVIDIA/dllogger.git@26a0f8f1958de2c0c460925ff6102a4d2486d6cc#egg=dllogger
7+
git+https://github.com/NVIDIA/dllogger@v0.1.0#egg=dllogger

PyTorch/SpeechSynthesis/Tacotron2/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@ numpy
33
inflect
44
librosa
55
scipy
6-
git+git://github.com/NVIDIA/dllogger.git@26a0f8f1958de2c0c460925ff6102a4d2486d6cc#egg=dllogger
6+
git+https://github.com/NVIDIA/dllogger@v0.1.0#egg=dllogger

PyTorch/Translation/GNMT/Dockerfile

+1-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ FROM ${FROM_IMAGE_NAME}
2424
ENV LANG C.UTF-8
2525
ENV LC_ALL C.UTF-8
2626

27-
RUN pip install --global-option="--cpp_ext" --global-option="--cuda_ext" git+git://github.com/NVIDIA/apex.git#egg=apex
27+
RUN pip install --global-option="--cpp_ext" --global-option="--cuda_ext" git+https://github.com/NVIDIA/apex.git#egg=apex
2828

2929
WORKDIR /workspace/gnmt
3030

PyTorch/Translation/GNMT/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@ pytablewriter==0.64.0
22
sacrebleu==1.2.10
33
sacremoses==0.0.19
44
pynvml==8.0.4
5-
git+git://github.com/rsennrich/subword-nmt.git@48ba99e657591c329e0003f0c6e32e493fa959ef
5+
git+https://github.com/rsennrich/subword-nmt.git@48ba99e657591c329e0003f0c6e32e493fa959ef

PyTorch/Translation/Transformer/Dockerfile

+1-2
Original file line numberDiff line numberDiff line change
@@ -44,5 +44,4 @@ RUN git clone https://github.com/rsennrich/subword-nmt.git /workspace/translatio
4444
RUN git clone https://github.com/NVIDIA/cutlass.git && cd cutlass && git checkout ed2ed4d6 && cd ..
4545
COPY . .
4646
RUN pip install -e .
47-
RUN pip install nvidia-pyindex
48-
RUN pip install nvidia-dllogger
47+
RUN pip install git+https://github.com/NVIDIA/[email protected]#egg=dllogger

TensorFlow/Recommendation/WideAndDeep/Dockerfile

-2
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,6 @@ 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 --no-cache-dir nvidia-pyindex
43-
4442
COPY requirements* ./
4543
RUN pip install --no-cache-dir -r requirements.txt
4644
RUN pip install --no-cache-dir --no-deps -r requirements-no-deps.txt
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
1-
nvidia-dllogger==0.1.0
1+
git+https://github.com/NVIDIA/dllogger@v1.0.0#egg=dllogger
22
ipdb==0.13.9
33
pyspark==2.3.1

TensorFlow/Segmentation/UNet_3D_Medical/Dockerfile

+1-2
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,7 @@ FROM ${FROM_IMAGE_NAME}
44
ADD . /workspace/unet3d
55
WORKDIR /workspace/unet3d
66

7-
RUN pip install nvidia-pyindex
8-
RUN pip install nvidia-dllogger==0.1.0
7+
RUN pip install git+https://github.com/NVIDIA/[email protected]#egg=dllogger
98
RUN pip install --disable-pip-version-check -r requirements.txt
109

1110
ENV TF_GPU_HOST_MEM_LIMIT_IN_MB=120000
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
git+git://github.com/NVIDIA/dllogger#egg=dllogger
1+
git+https://github.com/NVIDIA/dllogger#egg=dllogger

TensorFlow/Segmentation/VNet/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@ SimpleITK==1.1.0
22
requests
33
googledrivedownloader
44
tf2onnx
5-
git+git://github.com/NVIDIA/dllogger#egg=dllogger
5+
git+https://github.com/NVIDIA/dllogger#egg=dllogger
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
11
sacrebleu==1.2.10
2-
git+git://github.com/NVIDIA/dllogger.git@26a0f8f1958de2c0c460925ff6102a4d2486d6cc#egg=dllogger
2+
git+https://github.com/NVIDIA/dllogger@v0.1.0#egg=dllogger

TensorFlow2/Recommendation/WideAndDeep/Dockerfile

-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ ENV HOROVOD_CYCLE_TIME=0.1
2020
ENV HOROVOD_FUSION_THRESHOLD=67108864
2121
ENV HOROVOD_NUM_STREAMS=2
2222

23-
RUN pip install --no-cache-dir nvidia-pyindex
2423
ADD requirements.txt .
2524
RUN pip install --no-cache-dir -r requirements.txt
2625

Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
nvidia-dllogger==0.1.0
1+
git+https://github.com/NVIDIA/dllogger@v1.0.0#egg=dllogger

Tools/PyTorch/TimeSeriesPredictionPlatform/Dockerfile

+1-2
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,7 @@ ADD requirements.txt /workspace/requirements.txt
4444
ADD triton/requirements.txt /workspace/triton/requirements.txt
4545
RUN pip install -r /workspace/requirements.txt
4646
RUN pip install -r /workspace/triton/requirements.txt
47-
RUN pip install nvidia-pyindex
48-
RUN pip install nvidia-dllogger
47+
RUN pip install git+https://github.com/NVIDIA/dllogger#egg=dllogger
4948
RUN pip install --no-cache-dir -r requirements.txt -f https://data.dgl.ai/wheels/repo.html
5049

5150
# Add model files to workspace

0 commit comments

Comments
 (0)