diff --git a/pyproject.toml b/pyproject.toml index dcfe021c..1bd2cb95 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -71,8 +71,8 @@ dependencies = [ "simple-term-menu==1.5.2", "tabulate==0.9.0", "tensorboard>=2.14.1", - "torch==2.1.0", - "torchaudio==2.1.0", + "torch==2.3.1", + "torchaudio==2.3.1", "torchinfo==1.8.0", "typer>=0.12.4", "yaspin>=3.1.0", @@ -84,8 +84,8 @@ dynamic = ["version"] torch = [ # 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.3.1; sys_platform == "darwin"', + 'torchaudio==2.3.1; sys_platform == "darwin"', ] dev = [ "black~=24.3", diff --git a/requirements.torch.txt b/requirements.torch.txt index c24fab55..19f57add 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.3.1; sys_platform == "darwin" +torchaudio==2.3.1; sys_platform == "darwin" +torch==2.3.1+${CUDA_TAG}; sys_platform != "darwin" +torchaudio==2.3.1+${CUDA_TAG}; sys_platform != "darwin"