diff --git a/federated_learning/requirements.txt b/federated_learning/requirements.txt index ee010f71..be0929c2 100644 --- a/federated_learning/requirements.txt +++ b/federated_learning/requirements.txt @@ -1,7 +1,7 @@ --extra-index-url https://download.pytorch.org/whl/cu116 nvflare==2.0.10 -torch==1.12.1 +torch==1.13.0+cu116 torchvision==0.13.1 pandas==1.4.1 numpy==1.22.1