update colab notebook package installation

Summary:
to address: #1736

installation changed to:
```
!pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
!pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/nightly/cu121
!pip3 install torchmetrics==1.0.3
!pip3 install torchrec --index-url https://download.pytorch.org/whl/nightly/cu121
```

Reviewed By: henrylhtsang

Differential Revision: D54547391

fbshipit-source-id: ec2254eb7d4d2c2be4d696c59d9eeb37e25836d8

Author: iamzainhuda

Torchrec_Introduction.ipynb

@@ -19,98 +19,26 @@
       "source": [
         "## **Installation**\n",
         "Requirements:\n",
-        "- python >= 3.7\n",
+        "- python >= 3.9\n",
         "\n",
         "We highly recommend CUDA when using TorchRec. If using CUDA:\n",
-        "- cuda >= 11.0\n"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "BB2K68OYUJ_t"
-      },
-      "outputs": [],
-      "source": [
-        "# install conda to make installying pytorch with cudatoolkit 11.3 easier. \n",
-        "!wget https://repo.anaconda.com/miniconda/Miniconda3-py37_4.9.2-Linux-x86_64.sh\n",
-        "!chmod +x Miniconda3-py37_4.9.2-Linux-x86_64.sh\n",
-        "!bash ./Miniconda3-py37_4.9.2-Linux-x86_64.sh -b -f -p /usr/local"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "sFYvP95xaAER"
-      },
-      "outputs": [],
-      "source": [
-        "# install pytorch with cudatoolkit 11.6\n",
-        "!conda install pytorch pytorch-cuda=11.6 -c pytorch-nightly -c nvidia -y"
-      ]
-    },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "7iY7Uv11mJYK"
-      },
-      "source": [
+        "- cuda >= 11.8\n",
+        "\n",
         "Installing TorchRec will also install [FBGEMM](https://github.com/pytorch/fbgemm), a collection of CUDA kernels and GPU enabled operations to run "
       ]
     },
     {
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "tUnIw-ZREQJy"
-      },
-      "outputs": [],
-      "source": [
-        "# install torchrec\n",
-        "!pip3 install torchrec-nightly"
-      ]
-    },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "b6EHgotRXFQh"
-      },
-      "source": [
-        "The following steps are needed for the Colab runtime to detect the added shared libraries. The runtime searches for shared libraries in /usr/lib, so we copy over the libraries which were installed in /usr/local/lib/. **This is a very necessary step, only in the colab runtime**. "
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "_P45pDteRcWj"
-      },
-      "outputs": [],
-      "source": [
-        "!cp /usr/local/lib/lib* /usr/lib/"
-      ]
-    },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "n5_X2WOAYG3c"
-      },
-      "source": [
-        "\\**Restart your runtime at this point for the newly installed packages to be seen.** Run the step below immediately after restarting so that python knows where to look for packages. **Always run this step after restarting the runtime.**"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "8cktNrh8R9rC"
+        "id": "sFYvP95xaAER"
       },
       "outputs": [],
       "source": [
-        "import sys\n",
-        "sys.path = ['', '/env/python', '/usr/local/lib/python37.zip', '/usr/local/lib/python3.7', '/usr/local/lib/python3.7/lib-dynload', '/usr/local/lib/python3.7/site-packages']"
+        "!pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U\n",
+        "!pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/nightly/cu121\n",
+        "!pip3 install torchmetrics==1.0.3\n",
+        "!pip3 install torchrec --index-url https://download.pytorch.org/whl/nightly/cu121"
       ]
     },
     {
@@ -625,35 +553,46 @@
     }
   ],
   "metadata": {
-    "accelerator": "GPU",
-    "colab": {
-      "background_execution": "on",
-      "collapsed_sections": [],
-      "machine_shape": "hm",
-      "name": "Torchrec Introduction.ipynb",
-      "provenance": []
-    },
-    "interpreter": {
-      "hash": "d4204deb07d30e7517ec64733b2d65f24aff851b061e21418071854b06459363"
-    },
-    "kernelspec": {
-      "display_name": "Python 3.7.13 ('torchrec': conda)",
-      "language": "python",
-      "name": "python3"
-    },
-    "language_info": {
-      "codemirror_mode": {
-        "name": "ipython",
-        "version": 3
+    "custom": {
+      "cells": [],
+      "metadata": {
+        "accelerator": "GPU",
+        "colab": {
+          "background_execution": "on",
+          "collapsed_sections": [],
+          "machine_shape": "hm",
+          "name": "Torchrec Introduction.ipynb",
+          "provenance": []
+        },
+        "fileHeader": "",
+        "fileUid": "c9a29462-2509-4adb-a539-0318cf56bb00",
+        "interpreter": {
+          "hash": "d4204deb07d30e7517ec64733b2d65f24aff851b061e21418071854b06459363"
+        },
+        "isAdHoc": false,
+        "kernelspec": {
+          "display_name": "Python 3.7.13 ('torchrec': conda)",
+          "language": "python",
+          "name": "python3"
+        },
+        "language_info": {
+          "codemirror_mode": {
+            "name": "ipython",
+            "version": 3
+          },
+          "file_extension": ".py",
+          "mimetype": "text/x-python",
+          "name": "python",
+          "nbconvert_exporter": "python",
+          "pygments_lexer": "ipython3",
+          "version": "3.7.13"
+        }
       },
-      "file_extension": ".py",
-      "mimetype": "text/x-python",
-      "name": "python",
-      "nbconvert_exporter": "python",
-      "pygments_lexer": "ipython3",
-      "version": "3.7.13"
-    }
+      "nbformat": 4,
+      "nbformat_minor": 0
+    },
+    "indentAmount": 2
   },
   "nbformat": 4,
-  "nbformat_minor": 0
+  "nbformat_minor": 2
 }