Skip to content

Commit 5d2914e

Browse files
nv-kkudrynskinv-kkudrynski
committed
[GNMT/PyT] Updating for Ampere]
1 parent 21a77af commit 5d2914e

36 files changed

Lines changed: 591 additions & 618 deletions

‎PyTorch/Translation/GNMT/Dockerfile‎

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2018-2019, NVIDIA CORPORATION. All rights reserved.
1+
# Copyright (c) 2018-2020, NVIDIA CORPORATION. All rights reserved.
22
#
33
# Permission is hereby granted, free of charge, to any person obtaining a copy
44
# of this software and associated documentation files (the "Software"), to deal
@@ -18,15 +18,18 @@
1818
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
1919
# SOFTWARE.
2020

21-
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:19.05-py3
21+
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:20.06-py3
2222
FROM ${FROM_IMAGE_NAME}
2323

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
28+
2729
WORKDIR /workspace/gnmt
2830

2931
COPY requirements.txt .
3032
RUN pip install --no-cache-dir -r requirements.txt
33+
RUN pip install --no-cache-dir git+https://github.com/NVIDIA/dllogger.git#egg=dllogger
3134

3235
ADD . /workspace/gnmt

‎PyTorch/Translation/GNMT/LICENSE‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
Copyright (c) 2017 Elad Hoffer
2-
Copyright (c) 2018, NVIDIA CORPORATION. All rights reserved.
2+
Copyright (c) 2018-2020, NVIDIA CORPORATION. All rights reserved.
33

44
Permission is hereby granted, free of charge, to any person obtaining a copy
55
of this software and associated documentation files (the "Software"), to deal

‎PyTorch/Translation/GNMT/README.md‎

Lines changed: 374 additions & 191 deletions
Large diffs are not rendered by default.

‎PyTorch/Translation/GNMT/img/training_accuracy.png‎

-115 KB
Binary file not shown.

‎PyTorch/Translation/GNMT/launch.py‎

Lines changed: 0 additions & 307 deletions
This file was deleted.

‎PyTorch/Translation/GNMT/requirements.txt‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
1-
pytablewriter
1+
pytablewriter==0.47.0
22
sacrebleu==1.2.10
33
sacremoses==0.0.19
44
git+git://github.com/rsennrich/subword-nmt.git@48ba99e657591c329e0003f0c6e32e493fa959ef
5-
git+git://github.com/NVIDIA/apex.git#egg=apex

‎PyTorch/Translation/GNMT/scripts/docker/build.sh‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#!/bin/bash
22

3-
# Copyright (c) 2018-2019, NVIDIA CORPORATION. All rights reserved.
3+
# Copyright (c) 2018-2020, NVIDIA CORPORATION. All rights reserved.
44
#
55
# Permission is hereby granted, free of charge, to any person obtaining a copy
66
# of this software and associated documentation files (the "Software"), to deal
@@ -20,4 +20,4 @@
2020
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
2121
# SOFTWARE.
2222

23-
docker build . --rm -t gnmt:latest
23+
docker build . --network=host --rm -t gnmt:latest

‎PyTorch/Translation/GNMT/scripts/docker/interactive.sh‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#!/bin/bash
22

3-
# Copyright (c) 2018-2019, NVIDIA CORPORATION. All rights reserved.
3+
# Copyright (c) 2018-2020, NVIDIA CORPORATION. All rights reserved.
44
#
55
# Permission is hereby granted, free of charge, to any person obtaining a copy
66
# of this software and associated documentation files (the "Software"), to deal
@@ -20,4 +20,4 @@
2020
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
2121
# SOFTWARE.
2222

23-
nvidia-docker run --init -it --rm --ipc=host -v $PWD:/workspace/gnmt/ gnmt bash
23+
docker run --gpus all --init -it --rm --network=host --ipc=host -v $PWD:/workspace/gnmt/ gnmt bash

‎PyTorch/Translation/GNMT/scripts/filter_dataset.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2018-2019, NVIDIA CORPORATION. All rights reserved.
1+
# Copyright (c) 2018-2020, NVIDIA CORPORATION. All rights reserved.
22
#
33
# Permission is hereby granted, free of charge, to any person obtaining a copy
44
# of this software and associated documentation files (the "Software"), to deal

0 commit comments

Comments
 (0)