Skip to content

Latest commit

 

History

History
70 lines (47 loc) · 1.88 KB

File metadata and controls

70 lines (47 loc) · 1.88 KB

Tensorflow local GPU setup: Windows Native

Hardware Requirements

Note: TensorFlow binaries use AVX instructions which may not run on older CPUs.

Software Requirements

Create Environment

Open Anaconda Prompt.

conda create --name tf python=3.6
conda deactivate
conda activate tf

Install conda packages (Miniconda)

If you have installed Miniconda, follow the give commands

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
pip install --upgrade pip
pip install "tensorflow<2.11"

Installing conda packages (Anaconda)

  • Open Anaconda and select tf environment and install the following packages:
  • keras
  • tensorflow
  • tensorflow-gpu
  • The dependency packages will be installed automatically.

Verify the installation

  • Verify GPU setup:
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
  • Verify CPU setup:
python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

Verify with Jupyter Notebook

Download verify-tf and import inside tf environment jupyter notebook