keras使用tensorflow-gpu加速训练

需要VS2015运行时环境,不需要使用VS编译Simple什么的
一定要先去tensorflow看看gpu需要CUDA版本
在这里插入图片描述
1.下载CUDA,查看gpu是否兼容。
如果 nvcc -V 错误需要添加环境变量,自行百度。
添加环境变量后
打开DOS窗口 set PATH=C:
关闭DOS窗口。
再次打开DOS窗口,输入:echo %PATH% ,查看PATH 值已经生效。

2.下载cudnn,根据CUDA下载对应版本,解压,复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
在这里插入图片描述
3.安装tensorflow-gpu,根据CUDA下载对应版本
安装时使用清华镜像源

CUDA10.0
cudnn7.4
tensorflow-gpu1.14
keras2.15

查看gpu是否可用
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
import tensorflow as tf
print(tf.test.is_built_with_cuda())

在这里插入图片描述
csdn我是当做笔记使用,如果有人问就写详细点。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值