Add example notebook for docker (#2767)

* add example notebook for docker * remove unneeded file * update base image to NVIDIA PyTorch container * update --------- Co-authored-by: Chester Chen <[email protected]>
NVIDIA · Aug 24, 2024 · f2ef032 · f2ef032
1 parent a6e1100
commit f2ef032
Show file tree

Hide file tree

Showing 4 changed files with 521 additions and 0 deletions.
diff --git a/examples/advanced/docker/Dockerfile b/examples/advanced/docker/Dockerfile
@@ -0,0 +1,11 @@
+ARG PYTORCH_IMAGE=nvcr.io/nvidia/pytorch:24.07-py3
+FROM ${PYTORCH_IMAGE}
+
+ARG NVF_BRANCH=main
+
+RUN python3 -m pip install -U pip
+RUN python3 -m pip install -U setuptools
+RUN python3 -m pip install nvflare~=2.5.0rc
+
+WORKDIR /workspace/
+RUN git clone https://github.com/NVIDIA/NVFlare.git --branch ${NVF_BRANCH} --single-branch NVFlare
diff --git a/examples/advanced/docker/README.md b/examples/advanced/docker/README.md
@@ -0,0 +1,3 @@
+# NVIDIA FLARE with Docker
+
+The notebook in this directory walks through the creation and launch of Docker containers for NVIDIA FLARE.
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		# NVIDIA FLARE with Docker

		The notebook in this directory walks through the creation and launch of Docker containers for NVIDIA FLARE.