栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

TENSORFLOW2.1初学踩坑

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

TENSORFLOW2.1初学踩坑

教程:清华大学曹健老师 Tensorflow2.0教程

问题1:很多练习教程里老师可以每轮跑50000次,而我的只能跑1875次每轮。

最初可能是因为TensorFlow、CUDA、cudnn版本不匹配,因为按教程使用

conda install TensorFlow==2.1.0

时,无法找到2.1版本的,按评论区方法装了2.5版本

而2.5版本与教程的CUDA10.1版本不匹配。

 问题2:在装2.1后,出现(具体问题太多记不清楚了,根据当时的搜索记录可能是这几个)

  • Could not load dynamic library ‘cudart64_110.dll‘; dlerror: cudart64_110.dll not found
  • 出现 tensorflow/core/common_runtime/gpu/gpu_device.cc:1512] Adding visible gpu后卡很久
  • 或者GPU完全不动,运行代码显示无法找到GPU
  • Adding visible gpu devices: 0后卡住 

这时可以考虑卸载TensorFlow2.1 直接安装TensorFlowGPU版本

pip install tensorflow-gpu==2.1.0

切记选好版本,默认下载安装最新版,还要卸载重装。

附:CUDA可以通过下述链接直接安装不一定要按教程来。

CUDA Toolkit 10.1 original Archive | NVIDIA Developer

CUDA Toolkit 11.5 Downloads | NVIDIA Developer

Tensorflow-gpu安装超详细!!!_m0_49090516的博客-CSDN博客_tensorflowgpu安装

【踩坑教程】win10环境下RTX3050Ti安装Tensorflow-gpu2.5+CUDA11.2.1+cudnn8.1.0_typefree的博客-CSDN博客

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/339366.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号