tensorflow
fanqiliang630
这个作者很懒,什么都没留下…
展开
-
tensorflow不启用GPU
1. 环境确认tensorflow不启用GPU, 首先需要确认是否已经安装好了cuda, 如果没有安装好, 可以参考:https://blog.csdn.net/fanqiliang630/article/details/1146434032. 查看tf输出仔细查看tf的输出中是否有这一行:Not creating XLA devices, tf_xla_enable_xla_devices not set如果有, 则只需在主程序的开头加上如下代码:import oso..原创 2021-03-27 09:39:24 · 2063 阅读 · 2 评论 -
linux 下 cuda安装
1. 从nvidia官网下载nvdia驱动的.run文件和cudnn的tar包sudo chmod +x xx.runsudo ./xx.runsudo ./xx.run --uninstall # 卸载# cudnn的下载稍复杂, 需要登录, 且需要番茄才能有正常速度# 需要配置环境变量export LD_LIBRARY_PATH=${CUDNN_HOME}/lib64:$LD_LIBRARY_PATHexport CPLUS_INCLUDE_PATH=${CUDNN_HOME原创 2021-03-10 19:10:03 · 1639 阅读 · 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 · 1508 阅读 · 0 评论 -
tensorflow保存和加载训练进度
tf.train.Checkpoint : 用于创建checkpointtf.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 · 1585 阅读 · 0 评论