File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed
Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change 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
217223
218224######################################################################
219225# TorchRec Modules and Data Types
220- # ------------------------------
226+ # ----------------------------------
221227#
222228# This section goes over TorchRec Modules and data types including such
223229# entities as ``EmbeddingCollection`` and ``EmbeddingBagCollection``,
You can’t perform that action at this time.
0 commit comments