Tensorflow 2
小小和蓝胖
这个作者很懒,什么都没留下…
展开
-
Ubuntu 18.0.4配置Tensorflow 2.1.0 GPU环境 及 cuda&cuDNN版本查看(2020真实有效)
最近买了块Nvidia GeForce RTX 2070显卡,配置深度学习环境。环境是Ubuntu 18.0.4,Tensorflow 2.1.0,cuda 10.1.243,cuDNN 7.6.5。网上的资料基本全是陈旧的,查看cuDNN版本的部分全是错误的。因此记录备查(配置时间2020年6月24日)。 文章目录1. 安装显卡驱动2. 安装cuda和cuDNN3. 查看安装的cuda和cuDNN版本 1. 安装显卡驱动 从官网下载驱动 sudo init 3然后ctrl+alt+1进入字符模.原创 2020-06-24 13:33:55 · 678 阅读 · 7 评论 -
Tensorflow 2.1.0 中 keras.utils.plot_model()出错的解决方法(2020真实有效)
最近在学习Tensorflow 2,使用keras模型可视化的plot_model()绘制模型图的时候报错。上网查找解决方案,发现网上的解决方法没有切中核心,因此在此记录备查。 问题 希望绘制模型图: from tensorflow import keras keras.utils.plot_model(model, 'my_first_model.png') 报错: Failed to i...原创 2020-02-15 21:31:36 · 2494 阅读 · 4 评论