Window10安装Tensorflow2-GPU环境教程

Window10安装Tensorflow2-GPU环境教程

前提条件

Window10
Nvidia控制面板、系统信息、组件中CUDA >= 10.1（如下图所示，图中显卡为2080S）

vs_community 安装包
nvidia驱动程序 安装包 （如下图，按照自己的显卡的型号选择驱动程序。除此安装包自行在官网下载，其余皆在百度网盘中）

anconda 安装包
cuda-10.1 安装包
cudnn-10.1 安装包
pycharm 安装包

下载安装包

百度云盘链接
提取码：1yag
若失效请联系：jiangyu1181@163.com

安装顺序

1. vs_community
2. nvidia驱动程序
3. cuda
4. cudnn（分别将cudnn中的最后一级的文件放入cuda的安装目录中对应的文件夹，CUDA的默认的安装目录一般为 C:\Program Files\NVIDIA GPU Computing Toolkit）
5. pycharm

安装Tensorflow2.1-GPU环境

1. 打开Anaconda Prompt（按下键盘的win键，再输入an即可找到）
2. 添加conda、pip清华镜像源

// An highlighted block
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --set show_channel_urls yes

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pip -U
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

3. 创建安装TensorFlow2.1-GPU的虚拟环境

// An highlighted block
conda create -n tf2-gpu python=3.7 （记住虚拟环境安装的位置。tf2-gpu即为虚拟环境的名称）
codna install tensorflow-estimater==2.1.0（这步非常重要。若缺少这步，默认情况下在安装TensorFlow-gpu==2.1.0的时候会安装tensorflow-estimater==2.2.0，这会导致调用TensorFlow出错）
conda insatll tensorflow-gpu==2.1.0

4. 打开pycharm，配置已经安装的TensorFlow2.1虚拟环境