Install tensorflow-gpu and use it using Kernel in Jupyter

Pre-requisites

  1. Windows 10 OS.
  2. Anaconda installed
  3. Machine with CUDA supported GPU: Check if your Nvidia in this list.
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;%PATH%SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64;%PATH%SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include;%PATH%SET PATH=C:\tools\cuda\bin;%PATH%
conda create -n your_env_name python=3.7
conda info --envs
conda activate your_env_name
pip install ipykernel
python -m ipykernel install --user --name your_env_name \              --display-name disp_name
conda install tensorflow-gpu
conda install jupyter
Hello world for tensorflow

Extra Stuff!

conda env remove -n ENV_NAME
jupyter kernelspec list
jupyter kernelspec remove <kernel_name>

--

--

--

Love podcasts or audiobooks? Learn on the go with our new app.

Recommended from Medium

How to Deploy Django REST Framework and React-Redux application with Docker

Customer Story: Building A Voice Service With PhraseApp

Front-end best practices:

The Intersection of Crosswords and Code

A high quality software development life cycle explained easy

Embedding ads into flutter’s widget tree with admob_flutter

Experience at LetsGrowMore as Web Development Intern

Prometheus and Grafana with Persistent Storage

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
Sachinjose

Sachinjose

More from Medium

Protein structure prediction using AlphaFold2

Multi Class Text Classification using Python and GridDB | GridDB: Open Source Time Series Database…

Query MS Graph API in Python

How to create torch tensors