Skip to content

GitLab

Unverified Commit 6f72e71f authored by sshahrokhi's avatar sshahrokhi Committed by GitHub
Browse files

changing the requirements to a cpu torch version that works (#23483)

parent 5de2a6d5
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/flax/vision/requirements.txt
View file @ 6f72e71f
...@@ -3,6 +3,6 @@ jaxlib>=0.1.59 ...@@ -3,6 +3,6 @@ jaxlib>=0.1.59
flax>=0.3.5 flax>=0.3.5
optax>=0.0.8 optax>=0.0.8
-f https://download.pytorch.org/whl/torch_stable.html -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 -f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.10.0+cpu torchvision==0.12.0+cpu
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment