diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index bdef0da6..4eb8d0c7 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -1,5 +1,5 @@ -f https://download.pytorch.org/whl/torch/ -torch==2.2.2+cu121 +torch==2.6.0 -f https://download.pytorch.org/whl/torchvision/ torchvision==0.17.2+cu121 tensorboard diff --git a/requirements-torch-cxx11-abi.txt b/requirements-torch-cxx11-abi.txt index 81b897fe..b1e07c76 100644 --- a/requirements-torch-cxx11-abi.txt +++ b/requirements-torch-cxx11-abi.txt @@ -1,3 +1,3 @@ --extra-index-url https://download.pytorch.org/whl/ -torch==2.2.2+cpu.cxx11.abi +torch==2.6.0 tensorboard diff --git a/requirements-torch.txt b/requirements-torch.txt index bc863461..41d36b4b 100644 --- a/requirements-torch.txt +++ b/requirements-torch.txt @@ -1,6 +1,6 @@ --extra-index-url https://download.pytorch.org/whl/cpu/ torch==2.2.2+cpu ; sys_platform != 'darwin' torchvision==0.17.2+cpu ; sys_platform != 'darwin' -torch==2.2.2 ; sys_platform == 'darwin' +torch==2.6.0 ; sys_platform == 'darwin' torchvision==0.17.2 ; sys_platform == 'darwin' tensorboard