CUDA-enabled GPU

RubenMcCarty · Apr 3, 2019 · 7e94e0f · 7e94e0f
1 parent c39633e
commit 7e94e0f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/setup.ipynb b/setup.ipynb
@@ -174,6 +174,10 @@
     "\n",
     "```pip install tensorflow```\n",
     "\n",
+    "Note that if you have a CUDA-enabled GPU you should do instead\n",
+    "\n",
+    "```pip install tensorflow-gpu```\n",
+    "\n",
     "If you're using an older version of Anaconda, you might be better off with \n",
     "\n",
     "```conda install -c conda-forge tensorflow```\n",