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