![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
fanqiliang630
这个作者很懒,什么都没留下…
展开
-
tensorflow不启用GPU
1. 环境确认 tensorflow不启用GPU, 首先需要确认是否已经安装好了cuda, 如果没有安装好, 可以参考:https://blog.csdn.net/fanqiliang630/article/details/114643403 2. 查看tf输出 仔细查看tf的输出中是否有这一行: Not creating XLA devices, tf_xla_enable_xla_devices not set 如果有, 则只需在主程序的开头加上如下代码: import os o..原创 2021-03-27 09:39:24 · 2030 阅读 · 2 评论 -
linux 下 cuda安装
1. 从nvidia官网下载nvdia驱动的.run文件和cudnn的tar包 sudo chmod +x xx.run sudo ./xx.run sudo ./xx.run --uninstall # 卸载 # cudnn的下载稍复杂, 需要登录, 且需要番茄才能有正常速度 # 需要配置环境变量 export LD_LIBRARY_PATH=${CUDNN_HOME}/lib64:$LD_LIBRARY_PATH export CPLUS_INCLUDE_PATH=${CUDNN_HOME原创 2021-03-10 19:10:03 · 1623 阅读 · 0 评论 -
tensorflow2.4.0限制GPU使用的显存
physical_devices = tf.config.list_physical_devices('GPU') for dev in physical_devices: # 如果使用多块GPU时 tf.config.experimental.set_memory_growth(dev, True) 将上面代码放在程序的开头, 即可做到限制显存使用的效果.原创 2021-03-10 18:58:30 · 1495 阅读 · 0 评论 -
tensorflow保存和加载训练进度
tf.train.Checkpoint : 用于创建checkpoint tf.train.Checkpoint: tf.train.Checkpoint( root=None, **kwargs ) TensorFlow objects may containtrackablestate, such astf.Variables,tf.keras.optimizers.Optimizerimplementations,tf.data.Dataset iterators,tf....原创 2021-03-10 18:56:11 · 1568 阅读 · 0 评论