tomgoldstein / loss-landscape

Code for visualizing the loss landscape of neural nets
MIT License
2.72k stars 388 forks source link

installing dependencies (MPI /w h5py) #29

Open bhaprayan opened 3 years ago

bhaprayan commented 3 years ago

opening this issue for reference. i had to jump through some hurdles while trying to set up the dependencies for the repo.

to use MPI /w HDF5 follow this: https://docs.h5py.org/en/stable/mpi.html

that should fix things. if you're using conda, and get the error AttributeError: 'h5py.h5p.PropFAID' object has no attribute 'set_fapl_mpio' an additional step is:

conda install -c conda-forge "h5py>=2.9=mpi*"