初次尝试使用tensorflow-gpu训练模型

分类:人工智能 | 作者:凹凸曼 | 发表于2019/08/08 初次尝试使用tensorflow-gpu训练模型已关闭评论

要使用GPU来训练模型,首先用GPU-Z 工具看一下我的电脑显卡是否支持CUDA,查看的结果是支持 ,就用先python 环境加入 tensorflow-gpu模块

这个安装过程有点慢,并且有可能安装不成功,如果不行,就需要借助特殊网络。

一、安装完后,跑了一下训练模型是否支持,报错误,需要安装cuda-toolkit 


cuda-toolkit 下载地址:https://developer.nvidia.com/cuda-toolkit

我的环境是WINDOWS10+PYTHON 3.7.4  需要: cudnn-10.0-windows10-x64-v7.6.2.24 

安装好了 ,重启一下电脑 以防万一环境木有生效

二、又跑了一下训练模型

报错如下:ImportError: Could not find ‘cudnn64_7.dll’. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Note that installing cuDNN is a separate step from installing CUDA, and this DLL is often found in a different directory from the CUDA DLLs.

需要安装另外一个深度神经网络cudnn 库 


cudnn 下载地址:https://developer.nvidia.com/cudnn

下载这个需要注册,然后解压后把 cudnn64_7.dll 放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin   根据自己环境而定

三、再跑了一下训练模型,以为大功告成

木想到跑提示信息如下:

Ignoring visible gpu device (device: 0, name: GeForce GT 730, pci bus id: 0000:01:00.0, compute capability: 2.1) with Cuda compute capability 2.1. The minimum required Cuda capability is 3.5.

这个结果表明配置已经成功了,只是公司显卡计算能力不行 ,测试环境,只有用CPU讲究一下了。

文出自 “凹凸曼” 博客,请务必保留此出处 http://www.apoyl.com/?p=2515h

Tag:

日志信息 »

该日志于2019-08-08 16:53由 凹凸曼 发表在人工智能分类下, 评论已关闭。

目前盖楼

抱歉,评论被关闭

« »