diff --git a/intermediate_source/torchrec_intro_tutorial.py b/intermediate_source/torchrec_intro_tutorial.py index 75d2532670e..c4fdfc1833a 100644 --- a/intermediate_source/torchrec_intro_tutorial.py +++ b/intermediate_source/torchrec_intro_tutorial.py @@ -39,6 +39,12 @@ # !pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/cu121 # !pip3 install torchmetrics==1.0.3 # !pip3 install torchrec --index-url https://download.pytorch.org/whl/cu121 +# +# .. note:: +# If you are running this in Google Colab, make sure to switch to a GPU runtime type. +# For more information, +# see `Enabling CUDA `__ +# @@ -217,7 +223,7 @@ ###################################################################### # TorchRec Modules and Data Types -# ------------------------------ +# ---------------------------------- # # This section goes over TorchRec Modules and data types including such # entities as ``EmbeddingCollection`` and ``EmbeddingBagCollection``,