欢迎使用 FedML！

FedML Installation on NVIDIA Jetson Devices

  • Pull FedML RPI docker image
docker pull fedml/fedml:latest-nvidia-jetson-l4t-ml-r35.1.0-py3
  • Run Docker with "fedml login"
docker run -t -i --runtime nvidia fedml/fedml:latest-nvidia-jetson-l4t-ml-r35.1.0-py3 /bin/bash

root@8bc0de2ce0e0:/usr/src/app# fedml login 299

Install with pip

This method is only recommended to those who don't want to use docker. fedml needs to be installed without dependencies because Pytorch is not available in pip on Jetson.

  1. install Pytorch using python wheelsopen in new window
  2. install rest of the dependenciesopen in new window with pip3 except torch and torchvision:
  3. install fedml without dependencies since they are installed manually.
pip3 install fedml --no-dependencies

Issues and solutions:

  1. pip3 install h5py related error:
sudo apt-get install subversion
ln -s /usr/include/locale.h /usr/include/xlocale.h
sudo apt-get install libhdf5-serial-dev
  1. pip3 install sklearn related error:
sudo apt-get install build-essential libatlas-base-dev gfortran