Anaconda3 + CUDA10.1 + cudNN7.6 + Tensorflow2.1 环境配置

TensorFlow2.1 环境配置

1. 安装Anaconda

官网：https://anaconda.com

清华镜像：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

选择Anaconda3安装包文件下载

 

双击安装包，一路next（下一步），最后需要勾选“Add PATH....”的选项

2.安装CUDA10.1

进入Anaconda Prompt命令行界面（在开始菜单栏可找到）

首先创建tf2的conda环境

conda create -n tf2.1 python=3.6

会提示是否安装“Proceed([y]/n)?”

输入y，Enter回车，等待安装

接着输入安装CUDAtoolkit的命令

conda install cudatoolkit=10.1

Enter回车，等待安装

3.安装cudNN7.6

接着刚才的命令，输入

conda install cudnn=7.6

Enter回车，等待安装

4.安装ternsorflow2.1

接着刚才的命令，输入

pip install tensorflow==2.1

Enter回车，等待安装

等tensorflow安装完成之后，接着输入python

>>import tensorflow as tf

如果上述导入语句执行成功则说明环境安装成功，反之则未成功

>>tf.__version__

如果输出对应tensorflow版本号，如“2.1”，则说明环境配置成功