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

Install Terraform in Three Simple Steps

What is the difference between IaaS, PaaS and SaaS?

Everything you need to know about variables in Python

Magento Hosting: Advantages and Disadvantages Revealed

How to Begin Algorithmic Trading in Python

A maturity model for Web 3.0

Domain Transfer to another AWS Account

No-Index Log Management at S3 Scale

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

Use CUDA 11.0 for RAPIDS 21.12 with TensorFlow 2.4 in Ubuntu 18.04

Destroying Duck Hunt with OpenCV — image analysis for beginners

Using the MiniFrag Database to validate your SMARTS strings

From scratch to CUDA installation and TensorFlow compilation from the sources on Ubuntu 20.04