diff --git a/linux_requirements.txt b/linux_requirements.txt index 726ff5c..dda2ea3 100644 --- a/linux_requirements.txt +++ b/linux_requirements.txt @@ -5,7 +5,7 @@ h5py==3.8.0 gdown==4.7.1 pandas==2.0.2 --extra-index-url https://download.pytorch.org/whl/cu116 -torch==1.13.1+cu116 +torch==2.2.0 torchvision==0.14.1+cu116 torchaudio==0.13.1 -f https://data.pyg.org/whl/torch-1.13.0+cu116.html diff --git a/windows_requirements.txt b/windows_requirements.txt index 339b1c3..2ff8587 100644 --- a/windows_requirements.txt +++ b/windows_requirements.txt @@ -5,7 +5,7 @@ h5py==3.8.0 gdown==4.7.1 pandas==2.0.2 --extra-index-url https://download.pytorch.org/whl/cu116 -torch==1.13.1+cu116 +torch==2.2.0 torchvision==0.14.1+cu116 torchaudio==0.13.1 -f https://data.pyg.org/whl/torch-1.13.0+cu116.html