Tensorflow2.0-GPU安装教程

时间:2024-04-12 10:08:22

在安装tensorflow2.0版本时,踩了好多坑,希望借此机会整理一下。主要介绍在Win10系统下安装Tensorflow2.0Gpu的过程。

需要安装Anaconda3-显卡驱动-cuda-cudnn-tensorflowGpu2.0等五个部分,在安装过程中注意软件版本对应关系。如cuda10-cudnn10.0-tensorflowGpu2.0,所需软件如下。 

Tensorflow2.0-GPU安装教程

1. 下载Anaconda,链接https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

Tensorflow2.0-GPU安装教程Tensorflow2.0-GPU安装教程

安装成功后,配置Anaconda环境变量。右击电脑-高级系统设置-环境变量-用户变量-path,添加下图三个路径到环境变量。

Tensorflow2.0-GPU安装教程

检测anaconda3是否安装成功及版本信息:

打开Anaconda prompt 命令行输入 conda --version,回车,如下图所示。

Tensorflow2.0-GPU安装教程

2. 创建tensorflow2.0虚拟环境

conda create –n tensorflow2.0 python=3.6

若出现如下问题:

Tensorflow2.0-GPU安装教程

解决办法参考博客https://blog.csdn.net/ada0915/article/details/78529877

Tensorflow2.0-GPU安装教程

**环境:conda activate tensorflow2.0

查看当前环境已安装文件:pip list

升级pip方法:pip install --upgrade pip

查看所有虚拟环境:conda info --envs

Tensorflow2.0-GPU安装教程

3. 下载Cuda和Cudnn

(1)安装环境要求

Tensorflow2.0-GPU安装教程

注意:tensorflow-gpu2.0 只可以用cuda10.0版本

(2)更新显卡驱动

查看显卡驱动信息nvidia-smi,显示显卡驱动版本信息,在安装cuda时候,会自动安装驱动,不用主动更新驱动

(3)下载Cuda10.0版

下载后点击默认安装,cuda安装完后,配置环境变量

Tensorflow2.0-GPU安装教程

(4)下载Cudnn,链接https://developer.nvidia.com/rdp/form/cudnn-download-survey

需要注册账号,然后才能下载cudnn。Cudnn需要和cuda版本对应。

下载地址https://developer.nvidia.com/rdp/cudnn-archiveTensorflow2.0-GPU安装教程

下载完成后,解压。将cudnn-10.0-windows10-x64-v7.6.4.3文件夹下的bin,include,lib下的文件分别对应的复制到C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v10.0路径下的bin,include和lib下的文件夹下。

4. Tensorflow-Gpu安装

(1)安装tensorflow-gpu2.0先通过conda info --envs命令行查看创建虚拟的环境

(2)切换到tensorflow2.0的虚拟环境,安装tensorflow-gpu2.0

pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

也可以使用另一个镜像源

pip install tensorflow-gpu==2.0 -i https://pypi.douban.com/simple

(3)验证tensorflow是否安装成功:

Tensorflow2.0-GPU安装教程

如果出现版本信息,恭喜你,安装成功。