File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
pytorch/training/docker/2.7/py3/cu128
test/sagemaker_tests/huggingface_pytorch/training Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff 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/.
154154RUN 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 \
Original file line number Diff line number Diff line change @@ -9,12 +9,12 @@ docker-compose
99flake8 == 3.7.7
1010Flask == 1.1.1
1111mock
12- pytest
12+ pytest == 8.4.1
1313pytest-cov
1414pytest-rerunfailures
1515pytest-xdist
1616requests == 2.31.0
17- pluggy < 1 ,>= 0.3.0
17+ pluggy < 2 ,>= 0.3.0
1818requests_mock
1919fabric
2020invoke
You can’t perform that action at this time.
0 commit comments