抱歉,评论被关闭
初次尝试使用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