search

FedML-AI / FedML

FEDML - The unified and scalable ML library for large-scale distributed training, model serving, and federated learning. FEDML Launch, a cross-cloud scheduler, further enables running any AI jobs on any GPU cloud or on-premise cluster. Built on this library, TensorOpera AI (https://TensorOpera.ai) is your generative AI platform at scale.
https://TensorOpera.ai
Apache License 2.0
4.19k stars 786 forks source link

Error when running torch_fedavg_mnist_lr_one_line_example.py #480

Open Alan-JW opened 2 years ago

Alan-JW commented 2 years ago

Hi, @chaoyanghe, I encountered the following error when running torch_fedavg_mnist_lr_one_line_example.py, is this a version mismatch problem? image image

chaoyanghe commented 2 years ago

@Alan-JW please help to type "fedml env" and let us know your environment info.

Alan-JW commented 2 years ago

@chaoyanghe How should I output fedml env?

chaoyanghe commented 2 years ago

@Alan-JW in your fedml environement, launch a new shell window and type "fedml env". Here is the guidance: https://doc.fedml.ai/mlops/api.html

Alan-JW commented 2 years ago

@chaoyanghe ，This is my fedml environment！ image

fedml-dimitris commented 1 year ago

@Alan-JW Any progress on this, or you need additional help?