educational icon indicating copy to clipboard operation
educational copied to clipboard

Published 20 hours ago • verified publisher icon google-deepmind

pin the versions for the library installations

Open pdubz-sudo opened this issue 3 years ago • 1 comments

For the Introduction to Graph Neural Nets with JAX/jraph notebook in colab, I get an error AttributeError: module 'jaxlib.xla_extension' has no attribute 'PmapFunction'.

I didn't get this before but I'm guessing it's because the versions of the pip installs have changed. Would it be possible to pin the library versions in cell 1 so that the notebook always work even a year from now?

Screenshot from 2022-02-17 14-49-30

pdubz-sudo avatar Feb 17 '22 13:02 pdubz-sudo