安装 tensorflow-gpu-2.0.0版本并测试
前言:适合python 3.5-3.7,我的是python3.7.6 安装python的时候选择自动配置环境变量
步骤一:查看cuda和cudnn及tensorflow-gpu之间的对应关系
步骤二:安装CUDA 10.0 版本 一定要去官网下载
点击OK 就选择默认地址
选择同意
选择自定义
把VS勾掉不选
然后一直选下一步 直到 安装完成
然后开始配置环境变量
一、检查系统变量(两个都有)
二、添加环境变量
选path,编辑。。
三、cmd中输入nvcc -V测试
成功
步骤二: 安装对应的 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
完成!