From 57b5bbaf69e5f9907e35215c884997036c783ed9 Mon Sep 17 00:00:00 2001 From: sshahrokhi Date: Mon, 22 May 2023 09:58:55 -0700 Subject: [PATCH] changing the requirements to a cpu torch version that works (#23483) --- examples/flax/vision/requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/flax/vision/requirements.txt b/examples/flax/vision/requirements.txt index cf1859d7549477..539ffdc6fa9f74 100644 --- a/examples/flax/vision/requirements.txt +++ b/examples/flax/vision/requirements.txt @@ -3,6 +3,6 @@ jaxlib>=0.1.59 flax>=0.3.5 optax>=0.0.8 -f https://download.pytorch.org/whl/torch_stable.html -torch==1.9.0+cpu +torch==1.11.0+cpu -f https://download.pytorch.org/whl/torch_stable.html -torchvision==0.10.0+cpu \ No newline at end of file +torchvision==0.12.0+cpu