先去检查你TensorFlow对应的cuda和cudnn版本是否对应 

在 Windows 环境中从源代码构建  |  TensorFlow (google.cn)


打开C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin

先找到cudart64_100.dll文件(没有就去网上下载)

把文件复制到C:\Windows\System32,运行py程序发现第一个没问题了,但是后续文件找不到了

 同理,把找不到的文件全部从bin文件夹复制到System32文件夹下,运行发现

 问题解决

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐