diff --git a/requirements-tensorflow.txt b/requirements-tensorflow.txt index cf574331..f5bc37ef 100644 --- a/requirements-tensorflow.txt +++ b/requirements-tensorflow.txt @@ -1 +1 @@ -tensorflow~=2.20.0 +tensorflow~=2.21.0 diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index 7f614707..e55534f2 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -1,5 +1,6 @@ --f https://download.pytorch.org/whl/torch/ -torch==2.10.0+cu126 --f https://download.pytorch.org/whl/torchvision/ -torchvision==0.25.0+cu126 +--extra-index-url https://download.pytorch.org/whl/cu132 +--extra-index-url https://download.pytorch.org/whl/nightly/cu132 +--pre +torch==2.12 +torchvision==0.22 tensorboard diff --git a/requirements-torch.txt b/requirements-torch.txt index 97f036e5..03ded17e 100644 --- a/requirements-torch.txt +++ b/requirements-torch.txt @@ -1,6 +1,8 @@ ---extra-index-url https://download.pytorch.org/whl/cpu/ -torch==2.10.0+cpu ; sys_platform != 'darwin' -torchvision==0.25.0+cpu ; sys_platform != 'darwin' -torch==2.10.0 ; sys_platform == 'darwin' -torchvision==0.25.0 ; sys_platform == 'darwin' +--extra-index-url https://download.pytorch.org/whl/cpu +--extra-index-url https://download.pytorch.org/whl/nightly/cpu +--pre +torch==2.12 ; sys_platform != 'darwin' +torchvision==0.22 ; sys_platform != 'darwin' +torch==2.12 ; sys_platform == 'darwin' +torchvision>=0.22 ; sys_platform == 'darwin' tensorboard