From 022d03cb4a746146e2305dc6dfcad4d10fb876b3 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Thu, 3 Oct 2024 13:01:02 -0700 Subject: [PATCH 1/2] Update torchrec_intro_tutorial.py --- intermediate_source/torchrec_intro_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/intermediate_source/torchrec_intro_tutorial.py b/intermediate_source/torchrec_intro_tutorial.py index 75d2532670e..f8e7f0e360f 100644 --- a/intermediate_source/torchrec_intro_tutorial.py +++ b/intermediate_source/torchrec_intro_tutorial.py @@ -217,7 +217,7 @@ ###################################################################### # TorchRec Modules and Data Types -# ------------------------------ +# ---------------------------------- # # This section goes over TorchRec Modules and data types including such # entities as ``EmbeddingCollection`` and ``EmbeddingBagCollection``, From e179dc16b7c5898fec7291fd31028ff6dc9e1a0b Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Thu, 3 Oct 2024 13:04:23 -0700 Subject: [PATCH 2/2] Update torchrec_intro_tutorial.py --- intermediate_source/torchrec_intro_tutorial.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/intermediate_source/torchrec_intro_tutorial.py b/intermediate_source/torchrec_intro_tutorial.py index f8e7f0e360f..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 `__ +#