接外包,有相关需求的可以联系我:Telegram | Email

Tensorflow GPU 支持

该文章创建(更新)于07/28/2020,请注意文章的时效性!

缘由

最近把电脑重置后各种配置都要重新构建,最近又想玩玩Tensorflow就尝试安装,之前出过简单的教程win10安装tensorflow-gpu版本,但目前来看已经过时了,自己之前的代码都无法运行了。或许这就是被多数人吐槽的TF2和TF1完全就是不一样的东西吧?其他都还好说,最难的还是想搞GPU必须配置CUDA这一点了,这次Tensorflow上面的配置都没错,就是正式测试的时候发现少了一个文件!

安装

确定GPU是否支持CUDA

目前已知训练都是NVIDIA显卡或专业用来训练的卡,AMD没听说过可以用来训练!

  • 去官网搜索参数,或在桌面右击进入NVIDIA 控制面板如图中查看即可!

下载一个软件和压缩包

需注册账号下载!同时注意最新要支持的版本,我刚开始安装的是以前的10.0版本但始终报错,后来发现最新的TF要10.1以上才行!

配置环境变量

建议自己去查看一下是否加入,如没有则手动添加

SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;%PATH%
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\libx64;%PATH%
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include;%PATH%
SET PATH=C:\tools\cuda\bin;%PATH%

其他

报错缺少文件

2020-07-28 20:34:36.582508: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found
2020-07-28 20:34:36.588421: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.        
Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz
......................

  • 解决办法?
    下载该cudart64_101.dll后放入图中所示的文件夹中即可!

参考


要不赞赏一下?

微信
支付宝
PayPal
Bitcoin

版权声明 | Copyright

除非特别说明,本博客所有作品均采用知识共享署名-非商业性使用-禁止演绎 4.0 国际许可协议进行许可。转载请注明转自-
https://www.emperinter.info/2020/07/28/tensorflow-gpu-support/


要不聊聊?

我相信你准备留下的内容是经过思考的!【勾选防爬虫,未勾选无法留言】

*

*



微信公众号

优惠码

阿里云国际版20美元
Vultr10美元
搬瓦工 | Bandwagon应该有折扣吧?
域名 | namesiloemperinter(1美元)