diff --git a/requirements.torch.txt b/requirements.torch.txt index c24fab55..c3bbdc76 100644 --- a/requirements.torch.txt +++ b/requirements.torch.txt @@ -1,6 +1,6 @@ # these requirements have to be installed ahead of time in your environment and from a different URL: # CUDA_TAG=cu118 pip install -r requirements.torch.txt --find-links https://download.pytorch.org/whl/torch_stable.html -torch==2.1.0; sys_platform == "darwin" -torchaudio==2.1.0; sys_platform == "darwin" -torch==2.1.0+${CUDA_TAG}; sys_platform != "darwin" -torchaudio==2.1.0+${CUDA_TAG}; sys_platform != "darwin" +torch==2.2.0; sys_platform == "darwin" +torchaudio==2.2.0; sys_platform == "darwin" +torch==2.2.0+${CUDA_TAG}; sys_platform != "darwin" +torchaudio==2.2.0+${CUDA_TAG}; sys_platform != "darwin" diff --git a/requirements.txt b/requirements.txt index 78614124..43e14332 100644 --- a/requirements.txt +++ b/requirements.txt @@ -26,7 +26,7 @@ simple-term-menu==1.5.2 setuptools==59.5.0 # https://github.com/pytorch/pytorch/issues/69894 tabulate==0.9.0 tensorboard>=2.14.1 -torch==2.1.0 -torchaudio==2.1.0 +torch==2.2.0 +torchaudio==2.2.0 torchinfo==1.8.0 typer>=0.12.3