Merge pull request #422 from dchichkov/master

Adding instruction - how to execute in the Google Research Colaboratory

Author: ftynse

docs/source/installation_colab_research.rst

@@ -0,0 +1,41 @@
+Installation in the Google Colaboratory environment
+===================================================
+
+If you want to install TC in a Google Colaboratory environment, copy/paste and run
+the following code in the notebook. Please note, it will take 2-3 minutes to execute.
+
+Step 1: Create new Notebook in the Google Research Colaboratory
+^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+
+Open https://colab.research.google.com/ ,  create a new notebook and switch Runtime to GPU.
+
+Step 2: Create a new Code Cell, with the following code
+^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+
+.. code-block:: python
+
+    !wget -c https://repo.continuum.io/archive/Anaconda3-5.1.0-Linux-x86_64.sh
+    !chmod +x Anaconda3-5.1.0-Linux-x86_64.sh
+    !bash ./Anaconda3-5.1.0-Linux-x86_64.sh -b -f -p /usr/local
+    !conda install -q -y --prefix /usr/local -c pytorch -c tensorcomp tensor_comprehensions
+    
+    import sys
+    sys.path.append('/usr/local/lib/python3.6/site-packages/')
+    
+Step 3: Use TC normally, from Python/Torch environment
+^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+
+As an example, paste and execute the following code in a new Code Cell:
+
+.. code-block:: python
+
+    import tensor_comprehensions as tc
+    import torch
+    lang = """
+    def matmul(float(M, K) A, float(K, N) B) -> (C) {
+        C(m, n) +=! A(m, r_k) * B(r_k, n)
+    }
+    """
+    matmul = tc.define(lang, name="matmul")
+    mat1, mat2 = torch.randn(3, 4).cuda(), torch.randn(4, 5).cuda()
+    out = matmul(mat1, mat2)