Skip to content

Commit 56235fe

Browse files
authored
Merge branch 'master' into patch_2025_07_22
2 parents 7380518 + 4122f89 commit 56235fe

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

pytorch/training/docker/2.7/py3/cu128/Dockerfile.gpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ RUN mkdir -p /tmp/nvjpeg \
150150
# -l, --skip-limit-conf Skip EFA limit configuration
151151
# -k, --skip-kmod Skip EFA kmod installation
152152
# start from 0.38.0 EFA now bundles the AWS OFI NCCL plugin,
153-
# which can now be found in /opt/amazon/ofi-nccl rather than the original /opt/aws-ofi-nccl/.
153+
# which can now be found in /opt/amazon/ofi-nccl/lib/x86_64-linux-gnu rather than the original /opt/aws-ofi-nccl/.
154154
RUN mkdir /tmp/efa \
155155
&& cd /tmp/efa \
156156
&& curl -O https://s3-us-west-2.amazonaws.com/aws-efa-installer/aws-efa-installer-${EFA_VERSION}.tar.gz \

test/sagemaker_tests/huggingface_pytorch/training/requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,12 +9,12 @@ docker-compose
99
flake8==3.7.7
1010
Flask==1.1.1
1111
mock
12-
pytest
12+
pytest==8.4.1
1313
pytest-cov
1414
pytest-rerunfailures
1515
pytest-xdist
1616
requests==2.31.0
17-
pluggy<1,>=0.3.0
17+
pluggy<2,>=0.3.0
1818
requests_mock
1919
fabric
2020
invoke

0 commit comments

Comments
 (0)