torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Colab demo not working

Open devinbost opened this issue 11 months ago • 4 comments

When I try to walk through the steps of the Colab demo for Torchrec, I get this error:

image

Here is the link to the demo: https://colab.research.google.com/github/pytorch/torchrec/blob/main/Torchrec_Introduction.ipynb#scrollTo=4-v17rxkopQw

devinbost avatar Feb 28 '24 20:02 devinbost

@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.

IvanKobzarev avatar Feb 29 '24 15:02 IvanKobzarev

image

devinbost avatar Feb 29 '24 17:02 devinbost

@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.

IvanKobzarev avatar Feb 29 '24 18:02 IvanKobzarev

Thank you for the attention to this. Please let me know if there's anything I can do to help.

devinbost avatar Mar 15 '24 08:03 devinbost