Open rachelglenn opened 7 months ago
Hi @rachelglenn thanks for your interest!
Did you run prepare_data.sh first? (bash ./prepare_data.sh)
If your docker container can't connect to outside network, you can download the data before you start your container. And then mount the data directory.
Be sure to modify the data root in https://github.com/NVIDIA/NVFlare/blob/main/examples/hello-world/hello-pt/jobs/hello-pt/app/custom/cifar10trainer.py#L40 and https://github.com/NVIDIA/NVFlare/blob/main/examples/hello-world/hello-pt/jobs/hello-pt/app/custom/cifar10validator.py#L31
I would actually suggest you go through these examples first: https://github.com/NVIDIA/NVFlare/tree/main/examples/hello-world/ml-to-fl/pt
@IsaacYangSLA can you help with some insight ?
@rachelglenn Can you try to run the hello-numpy-sag example inside and see if it works as well?
I am trying to use a docker container to run examples in NVFLARE. I build the docker container by editing the one provided in the master branch of NFLARE.
https://github.com/NVIDIA/NVFlare/blob/main/docker/Dockerfile
I built the docker container and am running the container. I am trying to get the example hello-pt to run inside the docker container.
podman run --rm -it --security-opt label=disable --gpus all -p 8888:8888 --ulimit stack=67108864 --device nvidia.com/gpu=all -v /workspace/:/workspace localhost/nvflare/nvflare /bin/bash
nvflare simulator -w /tmp/nvflare/test -n 2 -t 2 /workspace/NVFlare_example/jobs/hello-pt
``