Update install-entrypoint.sh: add version lock for NCCL (#11475)

This commit is contained in:
Changming Sun 2022-05-10 15:37:55 -07:00 committed by GitHub
parent 3437967e63
commit 0ac2e6e546
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -27,4 +27,4 @@ if [ "${AUDITWHEEL_POLICY}" = "musllinux_1_1" ]; then
apk add --no-cache bash
fi
yum install -y yum-plugin-versionlock
yum versionlock cuda* libcudnn*
yum versionlock cuda* libcudnn* libnccl*