Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Colab demo not working #1736

Open
devinbost opened this issue Feb 28, 2024 · 4 comments
Open

Colab demo not working #1736

devinbost opened this issue Feb 28, 2024 · 4 comments

Comments

@devinbost
Copy link

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
@IvanKobzarev
Copy link
Contributor

@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
Copy link
Author

image

@IvanKobzarev
Copy link
Contributor

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

@zainhuda zainhuda mentioned this issue Mar 5, 2024
Open
zainhuda pushed a commit to zainhuda/torchrec that referenced this issue Mar 5, 2024
@kittenzcuddler43 @facebook-github-bot
update colab notebook package installation
f28dadf 
Summary: to address: pytorch#1736

Differential Revision: D54547391
facebook-github-bot pushed a commit that referenced this issue Mar 5, 2024
@kittenzcuddler43 @facebook-github-bot
update colab notebook package installation
c1f49f4 
Summary:
to address: #1736

installation changed to:
```
!pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
!pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/nightly/cu121
!pip3 install torchmetrics==1.0.3
!pip3 install torchrec --index-url https://download.pytorch.org/whl/nightly/cu121
```

Reviewed By: henrylhtsang

Differential Revision: D54547391

fbshipit-source-id: ec2254eb7d4d2c2be4d696c59d9eeb37e25836d8
@devinbost
Copy link
Author

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@IvanKobzarev @devinbost