build: manually update PyTorch version (#3568)

Set PyTorch and TorchVision version to nightly release 2024-08-04.

Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
pull/3601/head
Vivek Khandelwal 2024-08-06 21:36:39 +05:30 committed by GitHub
parent b1a232222f
commit 78d0fa8998
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 5 additions and 3 deletions

View File

@ -265,6 +265,8 @@ PY_BUILTIN_TO_TORCH_OP = {
"ge": torch.ops.aten.ge,
"ne": torch.ops.aten.ne,
"gt": torch.ops.aten.gt,
"mod": torch.ops.aten.fmod,
"eq": torch.ops.aten.eq,
}
# torch with cuda has a __version__ that looks like "2.1.0+cu113",

View File

@ -1 +1 @@
5147aeb49a367b4a338d446b604be4b65eed83f5
d6ea1eb2bc8ba770fd5a689a30e234837df27384

View File

@ -1,3 +1,3 @@
-f https://download.pytorch.org/whl/nightly/cpu/torch/
--pre
torch==2.5.0.dev20240718
torch==2.5.0.dev20240804

View File

@ -1,3 +1,3 @@
-f https://download.pytorch.org/whl/nightly/cpu/torchvision/
--pre
torchvision==0.20.0.dev20240718
torchvision==0.20.0.dev20240804