Я использовал следующую команду:
conda create -n TF2.1 python=3.7 conda activate TF2.1 conda install cudatoolkit=10.1 conda install cudnn=7.6 pip install tensorflow==2.1