当前位置:   article > 正文

Windows原生环境配置Tensorflow使用GPU_windows tensorflow 2.17 gpu

windows tensorflow 2.17 gpu

1 缘起

最近在折腾Transformer,选择Tensorflow作为基础框架,
距离上次应用Tensorflow跑项目已经有2年之多,同时,也换了电脑,
基础环境也没有了,开始重新搭建开发环境
此时,Tensorflow 2.x版本已经来到了2.17.0,
上上本书里使用的还是初代Tensorflow2.0.0,发展太快了,
再次使用Tensorflow调用GPU时,发现无法获取GPU设备
获取到的GPU列表为空,切换版本后出现异常,缺少动态链接库(DLL)。

于是,开始搜索,排查问题,最终,成功在Windows基础环境搭建Tensorflow GPU环境,即
Windows基础环境,Tensorflow可这正常使用GPU设备进行计算。

本文记录了完整的部署过程,帮助开发者基于Windows基础环境构建Tensorflow GPU开发环境。

Tensorflow正确获取GPU设备下图所示:

在这里插入图片描述

异常信息:

2024-07-21 00:18:29.455743: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library ‘cudart64_110.dll’; dlerror: cudart6

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/938900
推荐阅读
相关标签
  

闽ICP备14008679号