training_guideline: Added info about TF-GPU install

axondeepseg · Nov 12, 2018 · a7f96e3 · a7f96e3
1 parent fdeaa1e
commit a7f96e3
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/notebooks/training_guideline.ipynb b/notebooks/training_guideline.ipynb
@@ -9,7 +9,14 @@
     "In this notebook we will learn how to train a new model for axon & myelin segmentation. It covers the following scenario:\n",
     "\n",
     "* Train a model from scratch by defining the parameters of the network\n",
-    "* Make inference using the trained model\n"
+    "* Make inference using the trained model\n",
+    "\n",
+    "**Important:** If you have access to a GPU card, we strongly recommend you use it. To do so, TensorFlow needs to be uninstalled and then re-installed using the GPU mode:\n",
+    "```bash\n",
+    "source activate ads_venv  # Activate virtual environment. This line might depend on your ADS installation\n",
+    "pip uninstall tensorflow\n",
+    "pip install tensorflow-gpu==1.3.0\n",
+    "```"
    ]
   },
   {
@@ -368,7 +375,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.7"
+   "version": "3.6.6"
   }
  },
  "nbformat": 4,