Windows10下配置TensorFlow-GPU的详细过程

导读

参考了一些文章,挣扎了半天把这个环境配好了,分享一下过程。
三步骤
1. 安装anaconda
2. 安装cuda和cudnn
3. 下载TensorFlow-gpu和测试
说明: 本人的显卡:GeForce GTX 750 Ti

1. 安装anaconda

官网下载:https://www.anaconda.com/distribution
选择Windows下Python3.7(注意:必须选择64位,因为TF不支持Python32位的)
下载页面
下载完成后打开,然后就是傻瓜式的安装,一路next即可。

然后打开anaconda prompt命令界面,按照下面的代码创建一个虚拟环境

conda create tf-2.0      # 创建环境
activate tf-2.0          # 激活环境

2. 安装cuda 和cudnn

1)安装cuda
目前TensorFlow-GPU版最高只能用cuda-10.0的。
下载链接:
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
下载完成后,打开下载的驱动。安装路径默认就行。
取勾GeForce Experience
cuda安装界面
如果电脑上本身就有Visual Studio Integration,要将这个取消勾选,避免冲突了
cuda安装界面
点开Driver comonents,Display Driver这一行,前面显示的是Cuda本身包含的驱动版本是411.31
如果你电脑目前安装的驱动版本号新于Cuda本身自带的驱动版本号,那一定要把这个勾去掉。否则会安装失败(相同的话,就不用去取勾了)(其实这个失败不失败我不知道,本人装的时候是取消了。参考另外一篇文章的,链接在末尾)
在这里插入图片描述
接下来需要几分钟的时间安装,这是NVIDIA程序已完成的界面
cuda安装结束
打开此路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin,查看是否有nvcc.exe。有就说明CUDA安装已成功。
2) 安装cudnn
CDNN官网链接:
https://developer.nvidia.com/rdp/cudnn-download
选择cuDNN for CUDA10.0
需要注册一个账号,并且验证成功。网很重!!!
解压cuDNN到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

3)配置路径
查看CUDA环境路径
我的电脑——>属性——>高级系统设置——>环境变量
在系统变量中找到Path,双击打开添加两个路径:一是cudnn的bin路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\cudnn\bin,二是C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v10.0\extras\CUPTI\libx64

4) 测试成功与否
cmd进入命令提示符输入:nvcc - V(大写)
检查cuDNN

3. 下载TensorFlow-gpu和测试

1)下载TensorFlow-GPU
cmd打开命令提示符:

[1] activate tf-2.0
[2] pip install tensorflow-gpu==2.0.0-beta0
[3] pip install keras

等待安装完毕。

2) 测试
进入python(tf-2.0)环境(在命令提示符找到前面创建的环境),比如本人的创建路径:
训练创建的环境
进入之后,输入以下代码,验证是否安装成功:

import tensorflow as tf
tf.test.is_gpu_available()

如图显示说明安装成功:
检测安装成功

参考链接
link1
link2

更好的表述是生活一部分

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值