Skip to content

Commit e179dc1

Browse files
authored
Update torchrec_intro_tutorial.py
1 parent 022d03c commit e179dc1

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

intermediate_source/torchrec_intro_tutorial.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,12 @@
3939
# !pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/cu121
4040
# !pip3 install torchmetrics==1.0.3
4141
# !pip3 install torchrec --index-url https://download.pytorch.org/whl/cu121
42+
#
43+
# .. note::
44+
# If you are running this in Google Colab, make sure to switch to a GPU runtime type.
45+
# For more information,
46+
# see `Enabling CUDA <https://pytorch.org/tutorials/beginner/colab#enabling-cuda>`__
47+
#
4248

4349

4450

0 commit comments

Comments
 (0)