Skip to content

Commit

Permalink
[DLC][TNX] Freeze all aws-neuronx deps (#1397)
Browse files Browse the repository at this point in the history
  • Loading branch information
tosterberg authored Dec 14, 2023
1 parent f2bf595 commit 6108b5d
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions serving/docker/scripts/install_inferentia2.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,9 @@ echo "deb https://apt.repos.neuron.amazonaws.com ${VERSION_CODENAME} main" >/etc
curl -L https://apt.repos.neuron.amazonaws.com/GPG-PUB-KEY-AMAZON-AWS-NEURON.PUB | apt-key add -

# https://awsdocs-neuron.readthedocs-hosted.com/en/latest/release-notes/releasecontent.html#inf2-packages
apt-get update -y && apt-get install -y aws-neuronx-dkms=2.14.* \
aws-neuronx-collectives=2.18.* \
aws-neuronx-runtime-lib=2.18.* \
aws-neuronx-tools=2.15.*
apt-get update -y && apt-get install -y aws-neuronx-dkms=2.14.5.0 \
aws-neuronx-collectives=2.18.19.0* \
aws-neuronx-runtime-lib=2.18.15.0* \
aws-neuronx-tools=2.15.4.0

export PATH=/opt/aws/neuron/bin:$PATH

0 comments on commit 6108b5d

Please sign in to comment.