We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 022d03c commit e179dc1Copy full SHA for e179dc1
intermediate_source/torchrec_intro_tutorial.py
@@ -39,6 +39,12 @@
39
# !pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/cu121
40
# !pip3 install torchmetrics==1.0.3
41
# !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
48
49
50
0 commit comments