到目前为止,与GPU配合使用的最佳解决方案是使用以下方法安装张量流确定性:
pip install tensorflow-determinism
然后将以下代码添加到您的代码中
import tensorflow as tfimport osos.environ['TF_DETERMINISTIC_OPS'] = '1'
来源:https :
//github.com/NVIDIA/tensorflow-
determinism
欢迎分享,转载请注明来源:内存溢出
到目前为止,与GPU配合使用的最佳解决方案是使用以下方法安装张量流确定性:
pip install tensorflow-determinism
然后将以下代码添加到您的代码中
import tensorflow as tfimport osos.environ['TF_DETERMINISTIC_OPS'] = '1'
来源:https :
//github.com/NVIDIA/tensorflow-
determinism
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)