安装 tensorflow-gpu-2.0.0版本并测试

前言:适合python 3.5-3.7,我的是python3.7.6 安装python的时候选择自动配置环境变量
步骤一:查看cuda和cudnn及tensorflow-gpu之间的对应关系
安装 tensorflow-gpu-2.0.0版本并测试
步骤二:安装CUDA 10.0 版本 一定要去官网下载
安装 tensorflow-gpu-2.0.0版本并测试
点击OK 就选择默认地址

安装 tensorflow-gpu-2.0.0版本并测试
选择同意

安装 tensorflow-gpu-2.0.0版本并测试
选择自定义

安装 tensorflow-gpu-2.0.0版本并测试
把VS勾掉不选

安装 tensorflow-gpu-2.0.0版本并测试
然后一直选下一步 直到 安装完成

然后开始配置环境变量

一、检查系统变量(两个都有)
安装 tensorflow-gpu-2.0.0版本并测试
二、添加环境变量
选path,编辑。。
安装 tensorflow-gpu-2.0.0版本并测试
三、cmd中输入nvcc -V测试
安装 tensorflow-gpu-2.0.0版本并测试
成功

步骤二: 安装对应的 cudnn 7.4.2 还是去官网去下载

下载完后解压 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 这应该是默认地址 将解压后文件中的.dll 复制到v10.0文件中的bin文件。 把.h 文件复制到v10.0文件中的 include文件中 把 .lib文件复制到v10.0文件中lib文件中的x64文件中。

步骤三:下载tensorflow-gpu-2.0.0

在cmd中输入 pip3 install tensorflow-gpu==2.0.0

等待下载 下载成功后

测试

输入 python
按回车
输入 import tensorflow as tf
按回车
输入 tf.test.is_gpu_available()
按回车 等待 结果

结果为True
完成!安装 tensorflow-gpu-2.0.0版本并测试

相关文章: