Skip to content

Commit dd1f8dc

Browse files
authored
Fix Inf2 benchmark (#3177)
* Upgrade torch version * Test Inf2 benchmark fix * upgrade pip * Update benchmark test branch * Revert "upgrade pip" This reverts commit 7244fa5. * Revert "Update benchmark test branch" This reverts commit ccdd194. * Update benchmark test branch * Update bencmark model URLs to use S3 URL instead of alias * Revert "Test Inf2 benchmark fix" This reverts commit 47dc2d4. * Revert "Update benchmark test branch" This reverts commit b9a79fe. * Revert "Update bencmark model URLs to use S3 URL instead of alias" This reverts commit b11c473.
1 parent 5f5904b commit dd1f8dc

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

requirements/torch_neuronx_linux.txt

+4-5
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
--extra-index-url https://download.pytorch.org/whl/cpu
22
-r torch_common.txt
3-
torch==1.13.1+cpu
4-
torchvision==0.14.1+cpu
5-
torchtext==0.14.1
6-
torchaudio==0.13.1+cpu
7-
torchdata==0.5.1
3+
torch==2.1.2+cpu
4+
torchvision==0.16.2+cpu
5+
torchtext==0.16.2
6+
torchaudio==2.1.2+cpu

0 commit comments

Comments
 (0)