diff --git a/requirements-tensorflow.txt b/requirements-tensorflow.txt index 742e3f30..82f90bb1 100644 --- a/requirements-tensorflow.txt +++ b/requirements-tensorflow.txt @@ -1 +1 @@ -tensorflow~=2.13.0 +tensorflow~=2.16.2 diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index b69670fc..bdef0da6 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -1,5 +1,5 @@ -f https://download.pytorch.org/whl/torch/ -torch==2.0.1+cu117 +torch==2.2.2+cu121 -f https://download.pytorch.org/whl/torchvision/ -torchvision==0.15.2+cu117 +torchvision==0.17.2+cu121 tensorboard diff --git a/requirements-torch.txt b/requirements-torch.txt index fb417a31..bc863461 100644 --- a/requirements-torch.txt +++ b/requirements-torch.txt @@ -1,6 +1,6 @@ --extra-index-url https://download.pytorch.org/whl/cpu/ -torch==2.0.1+cpu ; sys_platform != 'darwin' -torchvision==0.15.2+cpu ; sys_platform != 'darwin' -torch==2.0.1 ; sys_platform == 'darwin' -torchvision==0.15.2 ; sys_platform == 'darwin' +torch==2.2.2+cpu ; sys_platform != 'darwin' +torchvision==0.17.2+cpu ; sys_platform != 'darwin' +torch==2.2.2 ; sys_platform == 'darwin' +torchvision==0.17.2 ; sys_platform == 'darwin' tensorboard