Skip to content

Commit 4901b56

Browse files
fix pt 2.7 nccl comment
1 parent 61a6b3e commit 4901b56

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
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 \

0 commit comments

Comments
 (0)