diff --git a/federated_learning/nvflare/nvflare_spleen_example/2-Client.ipynb b/federated_learning/nvflare/nvflare_spleen_example/2-Client.ipynb index a5813c7682..1bdb2fdd67 100644 --- a/federated_learning/nvflare/nvflare_spleen_example/2-Client.ipynb +++ b/federated_learning/nvflare/nvflare_spleen_example/2-Client.ipynb @@ -138,9 +138,9 @@ "cell_type": "markdown", "metadata": {}, "source": [ - "If you'd like to assigin GPU(s) for a client, please modify the script first. You may need to uncomment and edit the following line:\n", + "If you'd like to assigin GPU(s) for a client, please modify the script first. For example, if you'd like to assign GPU 1, the following line should be added into the script:\n", "\n", - "`# export CUDA_VISIBLE_DEVICES=`\n", + "`export CUDA_VISIBLE_DEVICES=1`\n", "\n", "[Please click here to edit for client 1](poc/site-1/startup/start.sh)\n", "\n", diff --git a/federated_learning/nvflare/nvflare_spleen_example/requirements.txt b/federated_learning/nvflare/nvflare_spleen_example/requirements.txt index 5eb8087a25..f66702b57e 100644 --- a/federated_learning/nvflare/nvflare_spleen_example/requirements.txt +++ b/federated_learning/nvflare/nvflare_spleen_example/requirements.txt @@ -1,6 +1,6 @@ pip setuptools -nvflare==2.0.2 +nvflare==2.0.16 monai==0.8.0 pytorch-ignite==0.4.6 tqdm==4.61.2