Tensorflow

Owned by Lokke A Highstein (Unlicensed)
Oct 04, 2023
1 min read

Tensorflow can be a bit tricky to install as it requires specific versions of cudatoolkit and CUDNN to be installed. As of this writing this works for version 2.12.1

replace "ENV_NAME" with the name of your conda environment

conda create -n ENV_NAME python=3.8
conda activate ENV_NAME
conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
conda install tensorflow=2.12.1=cuda112py38h4c2dfb8_0 -c conda-forge

If you need a different version of Tensorflow, please check the conda environment documentation here