torchrec
torchrec copied to clipboard
Colab demo not working
When I try to walk through the steps of the Colab demo for Torchrec, I get this error:
Here is the link to the demo: https://colab.research.google.com/github/pytorch/torchrec/blob/main/Torchrec_Introduction.ipynb#scrollTo=4-v17rxkopQw
@devinbost By default google colab has python of version 3.7. (I know we have this as minimal required version in the doc) I expect that this issue should go away with newer version of python like 3.10. I will try to check it for 3.10 updating python in colab.
@devinbost On the first screenshot the torch is installed in /usr/local/lib/python3.7/site-packages
The error you experienced did not happen for me on import torch for 3.10, but there is another error with missing fbgemm op jagged_2d_to_dense on import torchrec. I am debugging it now.
Thank you for the attention to this. Please let me know if there's anything I can do to help.